diff --git a/services/dask-sidecar/tests/unit/test_tasks.py b/services/dask-sidecar/tests/unit/test_tasks.py
index 932a1baa1d4c..8309c38cb532 100644
--- a/services/dask-sidecar/tests/unit/test_tasks.py
+++ b/services/dask-sidecar/tests/unit/test_tasks.py
@@ -524,7 +524,7 @@ def test_run_computational_sidecar_dask(
     worker_name = next(iter(dask_client.scheduler_info()["workers"]))
 
     output_data = future.result()
-    assert isinstance(output_data, dict)
+    assert isinstance(output_data, TaskOutputData)
 
     # check that the task produces expected logs
     worker_logs = [log for _, log in dask_client.get_worker_logs()[worker_name]]  # type: ignore