fix bug

Signed-off-by: invxp <invidentxp@gmail.com>