fix for issue #2
Merge pull request #4 from mprzybylak/issue2

fix for issue #2