Forgot one small check in h2o.gbm