diff --git a/R-package/R/predict.xgb.Booster.R b/R-package/R/predict.xgb.Booster.R
index cdb3f3dc1..15af9f34d 100644
--- a/R-package/R/predict.xgb.Booster.R
+++ b/R-package/R/predict.xgb.Booster.R
@@ -56,7 +56,7 @@ setMethod("predict", signature = "xgb.Booster",
       if (length(ret) == len){
           ret <- matrix(ret,ncol = 1)
       } else {
-          ret <- matrix(ret, ncol = nrow(newdata))
+          ret <- matrix(ret, ncol = len)
           ret <- t(ret)
       }
   }
diff --git a/R-package/demo/predict_leaf_indices.R b/R-package/demo/predict_leaf_indices.R
index 1fc64ba4a..480578c1d 100644
--- a/R-package/demo/predict_leaf_indices.R
+++ b/R-package/demo/predict_leaf_indices.R
@@ -12,7 +12,6 @@ nround = 5
 # training the model for two rounds
 bst = xgb.train(param, dtrain, nround, watchlist)
 cat('start testing prediction from first n trees\n')
-labels <- getinfo(dtest,'label')
 
 ### predict using first 2 tree
 pred_with_leaf = predict(bst, dtest, ntreelimit = 2, predleaf = TRUE)