repare bugs
Signed-off-by: 老牛啊 <obullxl@gmail.com>