small fix
Merge pull request #11 from zgao/master

small fix