Fixing broken unit test.