diff --git a/jvm-packages/xgboost4j/src/main/java/ml/dmlc/xgboost4j/java/XGBoost.java b/jvm-packages/xgboost4j/src/main/java/ml/dmlc/xgboost4j/java/XGBoost.java
index b2d0624c2..36ef18dbf 100644
--- a/jvm-packages/xgboost4j/src/main/java/ml/dmlc/xgboost4j/java/XGBoost.java
+++ b/jvm-packages/xgboost4j/src/main/java/ml/dmlc/xgboost4j/java/XGBoost.java
@@ -61,7 +61,7 @@ public class XGBoost {
       os.write(buf, 0, size);
     }
     in.close();
-    return Booster.loadModel(buf);
+    return Booster.loadModel(os.toByteArray());
   }
 
   /**