diff --git a/src/learner/evaluation-inl.hpp b/src/learner/evaluation-inl.hpp
index 14e66c5b7..32e7f7443 100644
--- a/src/learner/evaluation-inl.hpp
+++ b/src/learner/evaluation-inl.hpp
@@ -115,7 +115,6 @@ struct EvalCTest: public IEvaluator {
     utils::Check(preds.size() % info.labels.size() == 0,
                  "label and prediction size not match");
     size_t ngroup = preds.size() / info.labels.size() - 1;
-    ngroup = 1;
     const unsigned ndata = static_cast<unsigned>(info.labels.size());
     utils::Check(ngroup > 1, "pred size does not meet requirement");
     utils::Check(ndata == info.info.fold_index.size(), "need fold index");
diff --git a/src/tree/model.h b/src/tree/model.h
index f91e453f8..19524f8c9 100644
--- a/src/tree/model.h
+++ b/src/tree/model.h
@@ -288,6 +288,9 @@ class TreeModel {
                  "TreeModel: wrong format");
     utils::Check(fi.Read(&stats[0], sizeof(NodeStat) * stats.size()) > 0,
                  "TreeModel: wrong format");
+    if (param.size_leaf_vector != 0) {
+      utils::Check(fi.Read(&leaf_vector), "TreeModel: wrong format");
+    }
     // chg deleted nodes
     deleted_nodes.resize(0);
     for (int i = param.num_roots; i < param.num_nodes; i ++) {
@@ -308,6 +311,7 @@ class TreeModel {
     fo.Write(&param, sizeof(Param));
     fo.Write(&nodes[0], sizeof(Node) * nodes.size());
     fo.Write(&stats[0], sizeof(NodeStat) * nodes.size());
+    if (param.size_leaf_vector != 0) fo.Write(leaf_vector);
   }
   /*! 
    * \brief add child nodes to node
diff --git a/src/tree/param.h b/src/tree/param.h
index b1bfe69a3..52c273749 100644
--- a/src/tree/param.h
+++ b/src/tree/param.h
@@ -244,7 +244,6 @@ struct CVGradStats : public GradStats {
   }
   /*! \brief calculate gain of the solution */
   inline double CalcGain(const TrainParam &param) const {
-    return param.CalcGain(train[0].sum_grad, train[0].sum_hess);
     double ret = 0.0;
     for (unsigned i = 0; i < vsize; ++i) {
       ret += param.CalcGain(train[i].sum_grad,