another unit test fix