Updated failing unit test