diff --git a/src/lambo/utils/download.py b/src/lambo/utils/download.py
index e4fcc0f8f17f0d8763f7ce9cbec6fc1a1487a447..0aef6dfb767ec19e3711e32cfcab7031a604a626 100644
--- a/src/lambo/utils/download.py
+++ b/src/lambo/utils/download.py
@@ -16,8 +16,8 @@ logger = logging.getLogger(__name__)
 
 # The types of models available and their subdirectories in the model repository
 TYPE_TO_PATH = {
-    "LAMBO_no_pretraining": "vanilla",
-    "LAMBO": "full"}
+    "LAMBO_no_pretraining": "vanilla211-s",
+    "LAMBO": "full211-s"}
 
 # The adress of the remote repository
 _URL = "http://home.ipipan.waw.pl/p.przybyla/lambo/{type}/{treebank}.{extension}"