fixe bug

Signed-off-by: miaojian <kitemao@gmail.com>