From 2c2bde317809a870d12f49eb4c02e68b2b233ef1 Mon Sep 17 00:00:00 2001
From: Chris O'Hara <cohara87@gmail.com>
Date: Thu, 13 Jun 2024 11:51:45 +1000
Subject: [PATCH] Fix scheduler tests

---
 tests/dispatch/test_scheduler.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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,
         )