fix4 code change
Merge pull request #4 from jandi02/fix4

fix4 code change