update code of unit.c file

Signed-off-by: tristan_fei <tristan_fei@realsil.com.cn>