diff --git a/pybricks/tools/pb_module_tools.c b/pybricks/tools/pb_module_tools.c
index 14d4f5cd6..ae424792a 100644
--- a/pybricks/tools/pb_module_tools.c
+++ b/pybricks/tools/pb_module_tools.c
@@ -131,6 +131,9 @@ STATIC bool pb_module_tools_pbio_task_test_completion(mp_obj_t obj, uint32_t end
     pbio_task_t *task = MP_OBJ_TO_PTR(obj);
 
     // Keep going if not done yet.
+    if (task->status == PBIO_ERROR_AGAIN) {
+        pbio_task_run_once(task);
+    }
     if (task->status == PBIO_ERROR_AGAIN) {
         return false;
     }