Fixed test assertion