Fixed test to actually work