diff --git a/data/generate_dataset.py b/data/generate_dataset.py
index d3fabb3..9a9725f 100644
--- a/data/generate_dataset.py
+++ b/data/generate_dataset.py
@@ -147,7 +147,7 @@ def main(args):
                         Crise_size = rise_size_oi[enu]
                         Cblf = closest_baseline_freq[enu]
 
-                        rise_end_t = times_v[(idx_v > Ct_idx) & (fish_freq[id_idx] < Cblf + Crise_size * 0.37)]
+                        rise_end_t = times_v[(times_v[idx_v] > Ct_idx) & (fish_freq[id_idx] < Cblf + Crise_size * 0.37)]
                         if len(rise_end_t) == 0:
                             right_time_bound[enu] = np.nan
                         else: