fix error
fix up error
Merge pull request #6 from Yaliang/master

fix bug