Fixed test, hopefully