code cleanup

Signed-off-by: baron <baron@163.com>