Modify for Travis
Update test case