diff --git a/spark_/train_spark_GBT.java b/spark_/train_spark_GBT.java
index 8846130..53f590b 100644
--- a/spark_/train_spark_GBT.java
+++ b/spark_/train_spark_GBT.java
@@ -96,7 +96,7 @@ public static void main(String[] args) {
         // evaluate the model
 
         System.out.println("----------------- accuracy ----------------- "); 
-        System.out.println("Trained RF model:\n" + gbtModel.toDebugString());
+        System.out.println("Trained GBT model:\n" + gbtModel.toDebugString());
         System.out.println("accuracy: " + rmse );
         System.out.println("----------------- accuracy ----------------- "); 
     }