bug fix
refactor

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