Modify for travis-ci