bug fix

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