Fix some bugs.

Signed-off-by: Yunchao Chen <ytu_cyc@126.com>