diff --git a/src/autotrain/__init__.py b/src/autotrain/__init__.py
index 2f76635f44..331ef5ed9e 100644
--- a/src/autotrain/__init__.py
+++ b/src/autotrain/__init__.py
@@ -41,7 +41,7 @@
 warnings.filterwarnings("ignore", category=UserWarning, module="huggingface_hub")
 
 logger = Logger().get_logger()
-__version__ = "0.8.5.dev0"
+__version__ = "0.8.6.dev0"
 
 
 def is_colab():
diff --git a/src/autotrain/app/models.py b/src/autotrain/app/models.py
index a6e30544fb..0ea991a9f4 100644
--- a/src/autotrain/app/models.py
+++ b/src/autotrain/app/models.py
@@ -4,7 +4,7 @@
 
 
 def get_sorted_models(hub_models):
-    hub_models = [{"id": m.modelId, "downloads": m.downloads} for m in hub_models if m.private is False]
+    hub_models = [{"id": m.id, "downloads": m.downloads} for m in hub_models if m.private is False]
     hub_models = sorted(hub_models, key=lambda x: x["downloads"], reverse=True)
     hub_models = [m["id"] for m in hub_models]
     return hub_models