Fix for tests on travis