Fix tests with Travis