Fix another unit test failure