diff --git a/dan/manager/dataset.py b/dan/manager/dataset.py
index c888543c1291f786ab0a29a1ee37d10709f7b1ee..9d987c6827af887d9da6e6a6b1af6236a028e89b 100644
--- a/dan/manager/dataset.py
+++ b/dan/manager/dataset.py
@@ -139,7 +139,7 @@ class DatasetManager:
             )
 
     @staticmethod
-    def seed_worker():
+    def seed_worker(worker_id):
         worker_seed = torch.initial_seed() % 2**32
         np.random.seed(worker_seed)
         random.seed(worker_seed)