Fixed test expectation error