diff --git a/compare_gan/src/tfhub_models.ipynb b/compare_gan/src/tfhub_models.ipynb
index 4c562e1..f73cfb0 100644
--- a/compare_gan/src/tfhub_models.ipynb
+++ b/compare_gan/src/tfhub_models.ipynb
@@ -121,9 +121,10 @@
         "import numpy as np\n",
         "import pandas as pd\n",
         "\n",
-        "import tensorflow as tf\n",
+        "import tensorflow.compat.v1 as tf\n",
         "import tensorflow_hub as hub\n",
         "\n",
+        "tf.disable_v2_behavior()\n",
         "\n",
         "tf.logging.set_verbosity(tf.logging.ERROR)\n",
         "\n",
@@ -339,4 +340,4 @@
       "outputs": []
     }
   ]
-}
\ No newline at end of file
+}