diff --git a/colab_demo/faceswap-GAN_colab_demo.ipynb b/colab_demo/faceswap-GAN_colab_demo.ipynb
index 2f22c6f4..232ee95c 100644
--- a/colab_demo/faceswap-GAN_colab_demo.ipynb
+++ b/colab_demo/faceswap-GAN_colab_demo.ipynb
@@ -589,7 +589,7 @@
     "# VGGFace ResNet50\n",
     "#vggface = VGGFace(include_top=False, model='resnet50', input_shape=(224, 224, 3))'\n",
     "\n",
-    "from .vggface_models import RESNET50\n",
+    "from colab_demo.vggface_models import RESNET50\n",
     "vggface = RESNET50(include_top=False, weights=None, input_shape=(224, 224, 3))\n",
     "vggface.load_weights(\"rcmalli_vggface_tf_notop_resnet50.h5\")\n",
     "\n",