fixed issue #2 from dsully fork
Merge pull request #6 from dprelec/master

fixed issue #2 from dsully fork