Correcting a unit test