corrected UnitTest