Bugfix
Fixed #24 issue
Signed-off-by: mzcwlong <mzcwlong@vip.qq.com>