diff --git a/tests/dataset/dataset_download_test.py b/tests/dataset/dataset_download_test.py
index ca26acbe9..e3ca771fb 100644
--- a/tests/dataset/dataset_download_test.py
+++ b/tests/dataset/dataset_download_test.py
@@ -214,7 +214,7 @@ def test_dataset_extract_remove_finished_true(
 
     paths = pm.DatasetPaths(root=tmp_path, dataset='.')
     dataset = pm.Dataset(dataset_definition, path=paths)
-    dataset.extract(remove_finished=True, verbose=1)
+    dataset.extract(remove_finished=True, remove_top_level=False, verbose=1)
 
     mock_extract_archive.assert_has_calls([
         mock.call(
@@ -222,6 +222,7 @@ def test_dataset_extract_remove_finished_true(
             destination_path=tmp_path / 'raw',
             recursive=True,
             remove_finished=True,
+            remove_top_level=False,
             verbose=1,
         ),
     ])