diff --git a/official/vision/image_classification/common_test.py b/official/vision/image_classification/common_test.py
index 4c76813d35a..a6a967e8aa6 100644
--- a/official/vision/image_classification/common_test.py
+++ b/official/vision/image_classification/common_test.py
@@ -21,7 +21,6 @@
 import numpy as np
 import tensorflow as tf
 
-from absl import logging
 from tensorflow.python.platform import googletest
 from official.utils.misc import keras_utils
 from official.vision.image_classification import common
@@ -107,5 +106,5 @@ def _build_eval_output(self, top_1, eval_loss):
     return eval_output
 
 if __name__ == '__main__':
-  logging.set_verbosity(logging.ERROR)
+  tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)
   googletest.main()