fix bugs!

Signed-off-by: zhiying8710 <zhiying8710@hotmail.com>