fix bug
Merge pull request #3 from raypeng/master

fix bug