Updated some tests with H2O wrapper