diff --git a/tensorflow/python/kernel_tests/boosted_trees/resource_ops_test.py b/tensorflow/python/kernel_tests/boosted_trees/resource_ops_test.py
index a331b8670f7..250d5a403fc 100644
--- a/tensorflow/python/kernel_tests/boosted_trees/resource_ops_test.py
+++ b/tensorflow/python/kernel_tests/boosted_trees/resource_ops_test.py
@@ -16,10 +16,8 @@
 from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
-__p = print
-def _p(msg):
-  import sys
-  __p(msg, file=sys.stderr)
+import sys
+_p = print
 
 from google.protobuf import text_format