diff --git a/requirements.txt b/requirements.txt
index 9e5582d..1f87d20 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -10,4 +10,4 @@ timm~=0.9
 torch~=2.2
 onnx~=1.18
 onnxruntime~=1.22
-utils~=0.1.2.dev0
+utils~=0.1.0
diff --git a/scripts/dataset_building/data_gen.py b/scripts/dataset_building/data_gen.py
index 4f50fcb..43ccbb6 100644
--- a/scripts/dataset_building/data_gen.py
+++ b/scripts/dataset_building/data_gen.py
@@ -26,7 +26,7 @@ def generate_modulated_signals(output_dir: str) -> None:
     """
 
     for modulation in settings.modulation_types:
-        for snr in np.arange(settings.snr_start, settings.snr_end, settings.snr_step):
+        for snr in np.arange(settings.snr_start, settings.snr_stop, settings.snr_step):
             for i in range(3):
                 recording_length = settings.recording_length
                 beta = (