fx the bug
Merge pull request #12 from dinghaoDH/master

fx the bug