diff --git a/batchflow/models/torch/base.py b/batchflow/models/torch/base.py
index 54bd5372b..6196517e9 100755
--- a/batchflow/models/torch/base.py
+++ b/batchflow/models/torch/base.py
@@ -405,7 +405,7 @@ def callable_init(module): # example of a callable for init
 
     def __init__(self, config=None):
         if config is not None and not isinstance(config, (dict, Config)):
-            config = {'load/file': config}
+            config = {'load/path': config}
         self.model_lock = Lock()
 
         # Configs