Fix the tests for TravisCI