fix error when run unittest case