Fixing the broken unit test.