diff --git a/icu_benchmarks/data/split_process_data.py b/icu_benchmarks/data/split_process_data.py
index 9e4df484..54fa5c0e 100644
--- a/icu_benchmarks/data/split_process_data.py
+++ b/icu_benchmarks/data/split_process_data.py
@@ -198,7 +198,7 @@ def preprocess_data(
 
     # Generate cache
     if generate_cache:
-        caching(cache_dir, cache_file, data, load_cache)
+        caching(cache_dir, cache_file, sanitized_data, load_cache)
     else:
         logging.info("Cache will not be saved.")