fix bug

Signed-off-by: hancong03 <hancong03@baidu.com>