Fix test for Travis CI