Fix tests on Travis, hopefully