Fix test on Travis CI