Attempting to fix test that fails on Travis CI but not locally