[fix]issue 30

[wangtao-hengru]

根据issue：#30，根据注释修改了代码。