fix bug

Signed-off-by: tengattack <tengattack@foxmail.com>