Update for Travis CI tests