Modify for travis-CI