diff --git a/tests/unit_tests.py b/tests/unit_tests.py
index 5b204f0..cc5bbd4 100644
--- a/tests/unit_tests.py
+++ b/tests/unit_tests.py
@@ -11,10 +11,8 @@ import json
 import time
 
 import tensorflow as tf
-
-#-----------------------------------------
-
-tf_version = int(tf.__version__.split(".")[0])
+import logging
+tf.get_logger().setLevel(logging.ERROR)
 
 #-----------------------------------------