Fixed test that was only failing on CI