bug fix

Signed-off-by: Zhang Jing <rainius@163.com>