minor bug fix
Merge pull request #7 from maxziv/master

minor bug fix