diff --git a/rust/src/cost_scanner/token_activity.rs b/rust/src/cost_scanner/token_activity.rs
index 59f9b4bad7..7b9665dd5f 100644
--- a/rust/src/cost_scanner/token_activity.rs
+++ b/rust/src/cost_scanner/token_activity.rs
@@ -406,7 +406,7 @@ mod tests {
         let before = JsonlScanner::load_cache(ProviderId::Codex, Some(&shared));
         assert!(before.scan_since_key.is_some());
 
-        let keys = window_keys(Utc::now().date_naive(), TOKEN_ACTIVITY_DAYS);
+        let keys = window_keys(cost_bucket_zone().date(Utc::now()), TOKEN_ACTIVITY_DAYS);
         let activity = codex_activity_scanner(TOKEN_ACTIVITY_DAYS, Some(&shared))
             .with_sessions_dirs(vec![sessions]);
         let days = codex_token_activity(activity, &keys);