unit test corrected