corrected unit tests