Fix to Travis CI