diff --git a/climax/main/downscale_infer.py b/climax/main/downscale_infer.py
index 4f45992a21b6291b1e84f346a0ea392dda6c4b47..28dcfc19fcab84f769d832bceee4446027c0a9be 100644
--- a/climax/main/downscale_infer.py
+++ b/climax/main/downscale_infer.py
@@ -48,7 +48,7 @@ if __name__ == '__main__':
                 str(LOSS.min_amount).replace('.', ''),
                 repr(LOSS).strip('()')))
         else:
-            state_file = state_file.replace('.pt', '{}.pt'.format(
+            state_file = state_file.replace('.pt', '_{}.pt'.format(
                 repr(LOSS).strip('()')))
 
     # path to model state
diff --git a/climax/main/downscale_train.py b/climax/main/downscale_train.py
index 8ad352f676b155f8865bd0f6f8562544b3ddf69e..e0ccc3d0da192ee1b4ab98d6bda12a825a6b18d1 100644
--- a/climax/main/downscale_train.py
+++ b/climax/main/downscale_train.py
@@ -50,7 +50,7 @@ if __name__ == '__main__':
                 str(LOSS.min_amount).replace('.', ''),
                 repr(LOSS).strip('()')))
         else:
-            state_file = state_file.replace('.pt', '{}.pt'.format(
+            state_file = state_file.replace('.pt', '_{}.pt'.format(
                 repr(LOSS).strip('()')))
 
     # path to model state