bug fix

Signed-off-by: Demonsu <245681117@qq.com>