Corrected unit tests fail