revise unittest