diff --git a/src/features/battery_deltas.R b/src/features/battery_deltas.R
index 4074995f..63d092f0 100644
--- a/src/features/battery_deltas.R
+++ b/src/features/battery_deltas.R
@@ -15,7 +15,8 @@ if(nrow(battery) > 0){
               local_start_date_time = first(local_date_time),
               local_end_date_time = last(local_date_time),
               local_start_date = first(local_date),
-              local_end_date = last(local_date)) %>%
+              local_end_date = last(local_date),
+              local_day_segment = first(local_day_segment)) %>%
     select(-group_id) %>%
     filter(time_diff > 0.1) # Avoids including quick cycles
 } else {
@@ -24,7 +25,8 @@ if(nrow(battery) > 0){
                             local_start_date_time = character(),
                             local_end_date_time = character(),
                             local_start_date = character(),
-                            local_end_date = character())
+                            local_end_date = character(),
+                            local_day_segment = character())
 }
 
 write.csv(consumption, snakemake@output[[1]], row.names = FALSE)