diff --git a/climax/main/lr_range_test.py b/climax/main/lr_range_test.py
index 554734cedc93cf459d6735ec7aa3756dea785366..b498ea1f06a5b7c1f41f10a708ae8ddd4a09e851 100644
--- a/climax/main/lr_range_test.py
+++ b/climax/main/lr_range_test.py
@@ -66,7 +66,7 @@ if __name__ == '__main__':
         optim=OPTIM)
 
     # indicate lr range test
-    state_file = state_file.replace('.pt', 'lr_test.pt')
+    state_file = state_file.replace('.pt', '_lr_test.pt')
 
     # path to model state
     state_file = MODEL_PATH.joinpath(PREDICTAND, state_file)