code cleanup

Signed-off-by: yxjsolid <yxjsolid@gmail.com>