Modify for travis CI