corrected failing unit test