diff --git a/utils.py b/utils.py
index 3982b2f..0d1fda6 100755
--- a/utils.py
+++ b/utils.py
@@ -1,5 +1,9 @@
 import math
 import tensorflow as tf
+if tf.__version__.startswith("2."):
+    import tensorflow.compat.v1 as tf
+    tf.disable_eager_execution()
+    
 import numpy as np
 
 def asinh(x, scale=5.):