Fix Travis CI tests