Fixed failing unit test.