Fixed another unit test