bug fix
Merge pull request #1 from shumach217/iss2

bug fix