解决一个小BUG

Signed-off-by: dtysky <dtysky@outlook.com>