Fixed another unit test bug