fix for #1
Merge pull request #2 from luki3k5/1_fixing_the_problem

fix for #1