Fixing a unit test.