fix #7
Merge pull request #16 from minjiyang/iss7

fix #7