fix #343

之前提交的遗漏了一处问题，现修复