Modify for Travis CI