Fix bugs

Signed-off-by: dingxin <dingxin.git@gmail.com>