finetune unit test