diff --git a/internal/worker/state_awake.go b/internal/worker/state_awake.go
index d2a34db9..4ad6ae07 100644
--- a/internal/worker/state_awake.go
+++ b/internal/worker/state_awake.go
@@ -90,7 +90,11 @@ func (w *Worker) runStateAwake(ctx context.Context) {
 		}
 
 		// Do some rate limiting. This is mostly useful while developing.
-		time.Sleep(durationTaskComplete)
+		select {
+		case <-ctx.Done():
+			return
+		case <-time.After(durationTaskComplete):
+		}
 	}
 }