Corrected Unit Test