diff --git a/colab/GPU.ipynb b/colab/GPU.ipynb
index 296fb7fc9..8c08c5958 100644
--- a/colab/GPU.ipynb
+++ b/colab/GPU.ipynb
@@ -180,8 +180,8 @@
         "ii.addTask(\"Clone SillyTavern\", cloneTavern)\n",
         "ii.run()\n",
         "\n",
-        "from models import GetModels\n",
-        "model = GetModels(Version).get(Model, models.ModelData(Model, Version))\n",
+        "from models import GetModels, ModelData\n",
+        "model = GetModels(Version).get(Model, ModelData(Model, Version))\n",
         "\n",
         "# ---\n",
         "# KoboldAI\n",