diff --git a/src/samplers.py b/src/samplers.py
index 7723ae1..d6d79ef 100644
--- a/src/samplers.py
+++ b/src/samplers.py
@@ -64,7 +64,7 @@ def create_samplers(self, sample_config):
         idx_array = np.arange(self.dataset_size)
 
         # Shuffle array
-        np.random.shuffle(idx_array)
+        randomizer.shuffle(idx_array)
 
         # Get indices
         train_size = int(np.ceil(sample_config["train"] * self.dataset_size))