fixing another unit test