Fixed the last failing unit test.