diff --git a/tensorflow/contrib/learn/python/learn/estimators/random_forest.py b/tensorflow/contrib/learn/python/learn/estimators/random_forest.py
index 04d488987c8469..d2d464ca72c87c 100644
--- a/tensorflow/contrib/learn/python/learn/estimators/random_forest.py
+++ b/tensorflow/contrib/learn/python/learn/estimators/random_forest.py
@@ -317,6 +317,7 @@ def export(self,
              export_dir,
              input_fn,
              signature_fn=None,
+             input_feature_key=None,
              default_batch_size=1):
     """See BaseEstimator.export."""
     # Reset model function with basic device assigner.
@@ -330,7 +331,9 @@ def export(self,
         weights_name=self.weights_name)
     result = self._estimator.export(
         export_dir=export_dir,
-        use_deprecated_input_fn=True,
+        input_fn=input_fn,
+        input_feature_key=input_feature_key,
+        use_deprecated_input_fn=False,
         signature_fn=(signature_fn or
                       (export.regression_signature_fn
                        if self.params.regression else