diff --git a/DQN.ipynb b/DQN.ipynb
index d88f2c0..ab512c8 100755
--- a/DQN.ipynb
+++ b/DQN.ipynb
@@ -141,7 +141,8 @@
     "import os\n",
     "import random\n",
     "import gym\n",
-    "import tensorflow as tf\n",
+    "import tensorflow.compat.v1 as tf \n",
+    "tf.disable_v2_behavior() \n",
     "import numpy as np\n",
     "import imageio\n",
     "from skimage.transform import resize"