修改部分代码

Signed-off-by: neosfung <shuai.feng@intel.com>