corrected unit test