Fixing unit test.