Fixing failing unit test.