Fixed a regression on last commit
Merge pull request #4 from masylum/master

Fixed a regression on last commit