Corrected unit testing