Fix tests on travis