diff --git a/src/features/all_cleaning_individual/straw/main.py b/src/features/all_cleaning_individual/straw/main.py
index 8a222491..f132ee62 100644
--- a/src/features/all_cleaning_individual/straw/main.py
+++ b/src/features/all_cleaning_individual/straw/main.py
@@ -100,7 +100,7 @@ def straw_cleaning(sensor_data_files, provider):
         col.startswith('phone_screen_rapids_') or
         col.startswith('phone_wifi_visible')]
 
-    features[impute_zero] = impute(features[impute_zero], method="zero")
+    features[impute_zero] = features[impute_zero].fillna(0)
 
     ## (5) STANDARDIZATION 
     if provider["STANDARDIZATION"]:
diff --git a/src/features/all_cleaning_overall/straw/main.py b/src/features/all_cleaning_overall/straw/main.py
index b0baa760..9bb7bb3b 100644
--- a/src/features/all_cleaning_overall/straw/main.py
+++ b/src/features/all_cleaning_overall/straw/main.py
@@ -93,7 +93,8 @@ def straw_cleaning(sensor_data_files, provider, target):
         col.startswith('phone_messages_rapids_') or
         col.startswith('phone_screen_rapids_') or
         col.startswith('phone_wifi_visible')]
-    features[impute_zero] = impute(features[impute_zero], method="zero")
+        
+    features[impute_zero] = features[impute_zero].fillna(0)
 
     graph_bf_af(features, "5zero_imp")