diff --git a/tensorflow/compiler/tests/ensure_shape_op_test.py b/tensorflow/compiler/tests/ensure_shape_op_test.py
index 1729ffbae41..51aa82686fb 100644
--- a/tensorflow/compiler/tests/ensure_shape_op_test.py
+++ b/tensorflow/compiler/tests/ensure_shape_op_test.py
@@ -23,7 +23,6 @@ from tensorflow.python.framework import dtypes
 from tensorflow.python.framework import errors_impl
 from tensorflow.python.ops import array_ops
 from tensorflow.python.ops import check_ops
-from tensorflow.python.ops import math_ops
 from tensorflow.python.platform import test