Fix tests in Travis