code refined

Signed-off-by: songyang <waoywssy@gmail.com>