Fix bugs

Signed-off-by: LAN Xingcan <lanxingcan@gmail.com>