diff --git a/src/exo/worker/plan.py b/src/exo/worker/plan.py
index 70b2e02d..dacb3f33 100644
--- a/src/exo/worker/plan.py
+++ b/src/exo/worker/plan.py
@@ -9,6 +9,7 @@ from exo.shared.types.tasks import (
     ConnectToGroup,
     CreateRunner,
     DownloadModel,
+    ImageGeneration,
     LoadModel,
     Shutdown,
     StartWarmup,
@@ -265,7 +266,9 @@ def _pending_tasks(
 ) -> Task | None:
     for task in tasks.values():
         # for now, just forward chat completions
-        if not isinstance(task, ChatCompletion):
+        if not isinstance(task, ChatCompletion) and not isinstance(
+            task, ImageGeneration
+        ):
             continue
         if task.task_status not in (TaskStatus.Pending, TaskStatus.Running):
             continue