Corrected failing unit test