Update for Travis CI