diff --git a/tensorflow/contrib/boosted_trees/estimator_batch/estimator_test.py b/tensorflow/contrib/boosted_trees/estimator_batch/estimator_test.py
index de2d58bc895..60f92a0ff25 100644
--- a/tensorflow/contrib/boosted_trees/estimator_batch/estimator_test.py
+++ b/tensorflow/contrib/boosted_trees/estimator_batch/estimator_test.py
@@ -501,7 +501,7 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
             }
             nodes {
               dense_float_binary_split {
-                feature_column: 1
+                feature_column: 0
                 threshold: 0.52
                 left_id: 3
                 right_id: 4
@@ -512,7 +512,7 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
             }
             nodes {
               dense_float_binary_split {
-                feature_column: 1
+                feature_column: 0
                 threshold: 0.554
                 left_id: 5
                 right_id: 6
@@ -589,8 +589,7 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
         }
         nodes {
           dense_float_binary_split {
-            feature_column: 1
-            threshold: 0.52
+            threshold: 0.519999980927
             left_id: 3
             right_id: 4
           }
@@ -599,8 +598,7 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
         }
         nodes {
           dense_float_binary_split {
-            feature_column: 1
-            threshold: 0.554
+            threshold: 0.554000020027
             left_id: 5
             right_id: 6
           }
@@ -621,6 +619,13 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
             }
           }
         }
+        nodes {
+          leaf {
+            vector {
+              value: 0.0
+            }
+          }
+        }
         nodes {
           dense_float_binary_split {
             threshold: 1.0
@@ -631,13 +636,6 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
             gain: 0.888888895512
           }
         }
-        nodes {
-          leaf {
-            vector {
-              value: 0.0
-            }
-          }
-        }
         nodes {
           leaf {
             vector {
@@ -665,8 +663,7 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
         }
         nodes {
           dense_float_binary_split {
-            feature_column: 1
-            threshold: 0.52
+            threshold: 0.519999980927
             left_id: 3
             right_id: 4
           }
@@ -675,8 +672,7 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
         }
         nodes {
           dense_float_binary_split {
-            feature_column: 1
-            threshold: 0.554
+            threshold: 0.554000020027
             left_id: 5
             right_id: 6
           }
@@ -697,6 +693,13 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
             }
           }
         }
+        nodes {
+          leaf {
+            vector {
+              value: 0.0
+            }
+          }
+        }
         nodes {
           dense_float_binary_split {
             threshold: 1.0
@@ -707,13 +710,6 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
             gain: 0.727760672569
           }
         }
-        nodes {
-          leaf {
-            vector {
-              value: 0.0
-            }
-          }
-        }
         nodes {
           leaf {
             vector {
@@ -741,8 +737,7 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
         }
         nodes {
           dense_float_binary_split {
-            feature_column: 1
-            threshold: 0.52
+            threshold: 0.519999980927
             left_id: 3
             right_id: 4
           }
@@ -751,8 +746,7 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
         }
         nodes {
           dense_float_binary_split {
-            feature_column: 1
-            threshold: 0.554
+            threshold: 0.554000020027
             left_id: 5
             right_id: 6
           }