diff --git a/tests/dispatch/test_scheduler.py b/tests/dispatch/test_scheduler.py
index a88c4b7..554b633 100644
--- a/tests/dispatch/test_scheduler.py
+++ b/tests/dispatch/test_scheduler.py
@@ -449,7 +449,7 @@ def resume(
         poll = self.assert_poll(prev_output)
         input = Input.from_poll_results(
             main.__qualname__,
-            poll.coroutine_state,
+            any_unpickle(poll.typed_coroutine_state),
             call_results,
             Error.from_exception(poll_error) if poll_error else None,
         )