bug modify

Signed-off-by: Ghold <nhstljh302@gmail.com>