Fix tests on TravisCI