1 files changed, 1 insertions, 1 deletions
diff --git a/docs/mllib-decision-tree.md b/docs/mllib-decision-tree.md
index f31c4f8893..b5b454bc69 100644
--- a/docs/mllib-decision-tree.md
+++ b/docs/mllib-decision-tree.md
@@ -439,7 +439,7 @@ Double testMSE =
     public Double call(Double a, Double b) {
       return a + b;
     }
-  }) / data.count();
+  }) / testData.count();
 System.out.println("Test Mean Squared Error: " + testMSE);
 System.out.println("Learned regression tree model:\n" + model.toDebugString());