Fix a small bug.

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