fix a bug
Merge pull request #2 from einverne/master

fix a bug