fix issue
Merge pull request #13 from Atrae/fix_issue_11

fix issue