diff --git a/deep_q_network.py b/deep_q_network.py
index 1294f96..1a39291 100755
--- a/deep_q_network.py
+++ b/deep_q_network.py
@@ -1,7 +1,8 @@
 #!/usr/bin/env python
 from __future__ import print_function
 
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
+tf.disable_v2_behavior()
 import cv2
 import sys
 sys.path.append("game/")