Modify for travis ci