diff --git a/Utils.py b/Utils.py
index 4a7eb1c..21feb5d 100644
--- a/Utils.py
+++ b/Utils.py
@@ -84,9 +84,6 @@ class ProceesAudio():
         print(len(y), len(x))
         modified_segs = []
         prev_gains = np.ones(10)
-
-        max_part_amp = 0
-
         for segment, mod in zip(segments, y[0]):
             _, audios = split_stereo(segment=segment, max_amp=max_amp, sr=sr, num_parts=10)
             final_audios = []
@@ -114,7 +111,7 @@ class ProceesAudio():
 
 
         modified_segs = np.concatenate(modified_segs)
-        return modified_segs.astype(np.float32)
+        return modified_segs.astype(np.float32), y[0]
                 
     def get_training_data(self,  file_path, data_dir):
         for song in os.listdir(data_dir):