Fixed one failing unit test.