Corrected unit test