Fix another unit test error.