Fixed a bug in h2o.predict