diff --git a/tests/test_end_to_end.py b/tests/test_end_to_end.py
index 635c9a6..7d551d7 100644
--- a/tests/test_end_to_end.py
+++ b/tests/test_end_to_end.py
@@ -103,7 +103,7 @@ def test_ulmfit_fastai_bidir_end_to_end():
         name=lm_name,
     )
     exp.train_lm(num_epochs=1)
-    exp2 = ulmfit.train_clas.CLSHyperParams.from_lm(test_data / 'imdb', exp.model_dir)
+    exp2 = ulmfit.train_clas.CLSHyperParams.from_lm(str(test_data / 'imdb'), str(exp.model_dir))
     exp2.train_cls(num_lm_epochs=0, unfreeze=False, bs=4, )
 
 def test_ulmfit_moses_fa_bidir_end_to_end():
diff --git a/ulmfit/__main__.py b/ulmfit/__main__.py
index 5c1684a..23aec10 100644
--- a/ulmfit/__main__.py
+++ b/ulmfit/__main__.py
@@ -18,8 +18,8 @@ class ULMFiT:
 
     lm2 = LMHyperParams
     @wraps(CLSHyperParams)
-    def cls(self, dataset_path, baseon_path, **changes):
-        params = CLSHyperParams.from_lm(dataset_path, baseon_path, **changes)
+    def cls(self, dataset_path, base_lm_path, **changes):
+        params = CLSHyperParams.from_lm(dataset_path, base_lm_path, **changes)
         return FireView(train=params.train_cls)
 
 if __name__ == '__main__':
diff --git a/ulmfit/pretrain_lm.py b/ulmfit/pretrain_lm.py
index e68ed95..0bab350 100644
--- a/ulmfit/pretrain_lm.py
+++ b/ulmfit/pretrain_lm.py
@@ -243,6 +243,8 @@ class LMHyperParams:
 
     @classmethod
     def from_lm(cls, dataset_path, base_lm_path, **kwargs) -> 'LMHyperParams':
+        base_lm_path = Path(base_lm_path)
+        dataset_path = Path(dataset_path)
         with open(base_lm_path/'info.json', 'r') as f: d = json.load(f)
         d['dataset_path'] = dataset_path
         d['base_lm_path'] = base_lm_path