Dirty fix for #4

That's littlebit dirty fix fro #4
Merge pull request #5 from last-g/patch-1

Dirty fix for #4