Fix the tests

@liveh2o @localshred