diff --git a/data/generate_dataset.py b/data/generate_dataset.py
index caa01cf..b5d58a5 100644
--- a/data/generate_dataset.py
+++ b/data/generate_dataset.py
@@ -143,13 +143,11 @@ def main(args):
                     left_time_bound = times_v[rise_idx_oi]
                     right_time_bound = np.zeros_like(left_time_bound)
 
-                    for enu, Ct_idx in enumerate(times_v[rise_idx_oi]):
+                    for enu, Ct_oi in enumerate(times_v[rise_idx_oi]):
                         Crise_size = rise_size_oi[enu]
                         Cblf = closest_baseline_freq[enu]
 
-                        embed()
-                        quit()
-                        rise_end_t = times_v[(times_v[idx_v] > Ct_idx) & (fish_freq[id_idx] < Cblf + Crise_size * 0.37)]
+                        rise_end_t = times_v[(times_v > Ct_oi) & (fish_freq[id_idx] < Cblf + Crise_size * 0.37)]
                         if len(rise_end_t) == 0:
                             right_time_bound[enu] = np.nan
                         else: