diff --git a/keras/estimator/__init__.py b/keras/estimator/__init__.py
index 00d81a50ab4e0ff4f25d1583f12fae6158d16d33..71bb0e394a8f69d3f3fa30bb6669f90e424a3036 100644
--- a/keras/estimator/__init__.py
+++ b/keras/estimator/__init__.py
@@ -348,7 +348,7 @@ def model_to_estimator_v2(keras_model=None,
   """
 
   try:
-    from tensorflow_estimator.python.estimator import keras as keras_lib  # pylint: disable=g-import-not-at-top
+    from tensorflow_estimator.python.estimator import keras_lib  # pylint: disable=g-import-not-at-top
   except ImportError:
     raise NotImplementedError(
         'tf.keras.estimator.model_to_estimator function not available in your '
@@ -364,4 +364,4 @@ def model_to_estimator_v2(keras_model=None,
       use_v2_estimator=True,
       metric_names_map=metric_names_map,
       export_outputs=export_outputs)
-# LINT.ThenChange(//tensorflow_estimator/python/estimator/keras.py)
+# LINT.ThenChange(//tensorflow_estimator/python/estimator/keras_lib.py)