fix #251 Signed-off-by: yanxuean <yan.xuean@zte.com.cn>
Signed-off-by: Lantao Liu <lantaol@google.com>