fixed issue #18, 提交一下遗漏的代码