diff --git a/tensorflow/python/keras/estimator/__init__.py b/tensorflow/python/keras/estimator/__init__.py
index 79916aa6fc525a2d13b63b765289bfb8bc148f88..65a0e0c93ffec779dafe7557361cbec6896a565c 100644
--- a/tensorflow/python/keras/estimator/__init__.py
+++ b/tensorflow/python/keras/estimator/__init__.py
@@ -103,7 +103,8 @@ def model_to_estimator(
       custom_objects=custom_objects,
       model_dir=model_dir,
       config=config,
-      checkpoint_format=checkpoint_format)
+      checkpoint_format=checkpoint_format,
+      use_v2_estimator=False)
 
 
 @keras_export('keras.estimator.model_to_estimator', v1=[])
@@ -161,7 +162,6 @@ def model_to_estimator_v2(
       custom_objects=custom_objects,
       model_dir=model_dir,
       config=config,
-      checkpoint_format=checkpoint_format)
+      checkpoint_format=checkpoint_format,
+      use_v2_estimator=True)
 # LINT.ThenChange(//tensorflow_estimator/python/estimator/keras.py)
-
-