Fixed another unit test.