diff --git a/deeplabcut/create_project/modelzoo.py b/deeplabcut/create_project/modelzoo.py
index 6a431a03f6..31bafb84d6 100644
--- a/deeplabcut/create_project/modelzoo.py
+++ b/deeplabcut/create_project/modelzoo.py
@@ -261,6 +261,7 @@ def create_pretrained_project(
         download_huggingface_model(model, train_dir)
 
         pose_cfg = deeplabcut.auxiliaryfunctions.read_plainconfig(path_train_config)
+        pose_cfg["dataset_type"] = "imgaug"
         print(path_train_config)
         # Updating config file:
         dict_ = {
diff --git a/examples/COLAB/COLAB_DLC_ModelZoo.ipynb b/examples/COLAB/COLAB_DLC_ModelZoo.ipynb
index 705fc4ad69..d8fb77222b 100644
--- a/examples/COLAB/COLAB_DLC_ModelZoo.ipynb
+++ b/examples/COLAB/COLAB_DLC_ModelZoo.ipynb
@@ -3,8 +3,8 @@
     {
       "cell_type": "markdown",
       "metadata": {
-        "id": "view-in-github",
-        "colab_type": "text"
+        "colab_type": "text",
+        "id": "view-in-github"
       },
       "source": [
         "<a href=\"https://colab.research.google.com/github/DeepLabCut/DeepLabCut/blob/master/examples/COLAB/COLAB_DLC_ModelZoo.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
@@ -56,22 +56,17 @@
       "source": [
         "#click the play icon (this will take a few minutes to install all the dependencies!)\n",
         "#Install the latest DeepLabCut version from master:\n",
-        "!pip install https://github.com/DeepLabCut/DeepLabCut/archive/master.zip\n",
-        "!pip install 'deeplabcut[tf,modelzoo]'==2.3.0\n",
-        "!pip install huggingface_hub\n",
-        "%reload_ext numpy\n",
-        "%reload_ext matplotlib\n",
-        "%reload_ext mpl_toolkits"
+        "!pip install \"deeplabcut[tf,modelzoo] @ git+https://github.com/DeepLabCut/DeepLabCut.git@master\""
       ]
     },
     {
       "cell_type": "markdown",
-      "source": [
-        "###proTip: be sure to click \"restart runtime button\" if it appears above ^ "
-      ],
       "metadata": {
         "id": "zYm6DljQB0Y7"
-      }
+      },
+      "source": [
+        "###proTip: be sure to click \"restart runtime button\" if it appears above ^ "
+      ]
     },
     {
       "cell_type": "markdown",
@@ -92,10 +87,6 @@
       "outputs": [],
       "source": [
         "import os\n",
-        "\n",
-        "# stifle tensorflow warnings, like we get it already.\n",
-        "os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'\n",
-        "\n",
         "import deeplabcut"
       ]
     },
@@ -295,11 +286,12 @@
   ],
   "metadata": {
     "colab": {
+      "include_colab_link": true,
       "name": "Copy of COLAB_DLC_ModelZoo.ipynb",
       "provenance": [],
-      "toc_visible": true,
-      "include_colab_link": true
+      "toc_visible": true
     },
+    "gpuClass": "standard",
     "kernelspec": {
       "display_name": "Python 3",
       "language": "python",
@@ -316,8 +308,7 @@
       "nbconvert_exporter": "python",
       "pygments_lexer": "ipython3",
       "version": "3.7.7"
-    },
-    "gpuClass": "standard"
+    }
   },
   "nbformat": 4,
   "nbformat_minor": 0