Fix tests on travis ci.