fix bug

Signed-off-by: zhouyou <zhouyoug@gmail.com>