diff --git a/skyllh/core/signal_generator.py b/skyllh/core/signal_generator.py
index 9da2049442..c2694cbf1e 100644
--- a/skyllh/core/signal_generator.py
+++ b/skyllh/core/signal_generator.py
@@ -699,7 +699,8 @@ def _draw_valid_sig_events_for_dataset_and_shg(
             )
             m = (events_meta['ds_idx'] == ds_idx) &\
                 (events_meta['shg_idx'] == shg_idx)
-            events = mc[events_meta['ev_idx'][m]]
+            events_meta = events_meta[m]
+            events = mc[events_meta['ev_idx']]
             if len(events) > 0:
                 events = shg.sig_gen_method.\
                     signal_event_post_sampling_processing(