Fixing the unit test.