modify a bug

Signed-off-by: miaojian <kitemao@gmail.com>
Merge pull request #40 from kitemao/master

modify a bug