Fixing this unit test.