diff --git a/unit_tests/sources/declarative/async_job/test_integration.py b/unit_tests/sources/declarative/async_job/test_integration.py
index 36814a50..a0b6195b 100644
--- a/unit_tests/sources/declarative/async_job/test_integration.py
+++ b/unit_tests/sources/declarative/async_job/test_integration.py
@@ -38,7 +38,7 @@
 
 class MockAsyncJobRepository(AsyncJobRepository):
     def start(self, stream_slice: StreamSlice) -> AsyncJob:
-        return AsyncJob("a_job_id", StreamSlice(partition={}, cursor_slice={}))
+        return AsyncJob("a_job_id", stream_slice)
 
     def update_jobs_status(self, jobs: Set[AsyncJob]) -> None:
         for job in jobs:
diff --git a/unit_tests/sources/declarative/partition_routers/test_async_job_partition_router.py b/unit_tests/sources/declarative/partition_routers/test_async_job_partition_router.py
index b357acfb..ccc57cc9 100644
--- a/unit_tests/sources/declarative/partition_routers/test_async_job_partition_router.py
+++ b/unit_tests/sources/declarative/partition_routers/test_async_job_partition_router.py
@@ -78,5 +78,7 @@ def test_stream_slices_with_parent_slicer():
         attempts_per_job = list(partition.jobs)
         assert len(attempts_per_job) == 1
         assert attempts_per_job[0].api_job_id() == "a_job_id"
-        assert attempts_per_job[0].job_parameters() == StreamSlice(partition={}, cursor_slice={})
+        assert attempts_per_job[0].job_parameters() == StreamSlice(
+            partition={"parent_id": str(i)}, cursor_slice={}
+        )
         assert attempts_per_job[0].status() == AsyncJobStatus.COMPLETED