diff --git a/tensorflow/lite/testing/op_tests/fused_batch_norm.py b/tensorflow/lite/testing/op_tests/fused_batch_norm.py
index a884597731c..10ed48ed013 100644
--- a/tensorflow/lite/testing/op_tests/fused_batch_norm.py
+++ b/tensorflow/lite/testing/op_tests/fused_batch_norm.py
@@ -44,7 +44,7 @@ def make_fused_batch_norm_tests(options):
     variance = create_tensor_data(parameters["dtype"], scale_shape)
 
     x = create_tensor_data(parameters["dtype"], parameters["input_shape"])
-    [x_norm, _, _] = tf.nn.fused_batch_norm(
+    [x_norm, _, _] = tf.compat.v1.nn.fused_batch_norm(
         x,
         scale,
         offset,