diff --git a/mu_map/training/random_search.py b/mu_map/training/random_search.py
index f2afd1046ebb316fce2c06093402b489e7fb2dca..e6b57b3bcfdaf438c08d6fec6e8141dd562beb12 100644
--- a/mu_map/training/random_search.py
+++ b/mu_map/training/random_search.py
@@ -290,8 +290,8 @@ class RandomSearchCGAN(RandomSearch):
 
             seed = random.randint(0, 2**32 - 1)
             random.seed(seed)
-            torch.manual_seed(args.seed)
-            np.random.seed(args.seed)
+            torch.manual_seed(seed)
+            np.random.seed(seed)
             self.logger.info(f"Random seed for iteration {i} is {seed}")
 
             self._setup_run(i)
@@ -453,5 +453,5 @@ class RandomSearchCGAN(RandomSearch):
 
 
 if __name__ == "__main__":
-    random_search = RandomSearchCGAN(iterations=4)
+    random_search = RandomSearchCGAN(iterations=10)
     random_search.run()