Updated a test that was failing on travis.