diff --git a/climax/main/lr_range_test.py b/climax/main/lr_range_test.py
index 1bdaa16045b05a7fefa76968a888b7490027a6a5..805789550d097f3cc95a1af9f55949ab060dd097 100644
--- a/climax/main/lr_range_test.py
+++ b/climax/main/lr_range_test.py
@@ -27,7 +27,7 @@ from climax.main.config import (ERA5_PLEVELS, ERA5_PREDICTORS, PREDICTAND,
                                 CALIB_PERIOD, DOY, SHUFFLE, BATCH_SIZE, OPTIM,
                                 NORM, NET, LOSS, FILTERS, DEM, DEM_FEATURES,
                                 STRATIFY, WET_DAY_THRESHOLD, VALID_SIZE,
-                                ANOMALIES)
+                                ANOMALIES, CHUNKS)
 from climax.main.io import ERA5_PATH, OBS_PATH, DEM_PATH, MODEL_PATH
 
 # module level logger