Fix unittest for DecisionStump.