diff --git a/src/features/utils/utils.py b/src/features/utils/utils.py
index 9e347284..b0399de8 100644
--- a/src/features/utils/utils.py
+++ b/src/features/utils/utils.py
@@ -1,6 +1,7 @@
 rapids_log_tag =  "RAPIDS:"
 
 def filter_data_by_segment(data, time_segment):
+    data.dropna(subset=["assigned_segments"], inplace=True)
     if(data.shape[0] == 0): # data is empty
         data["local_segment"] = data["timestamps_segment"] = None
         return data