Fix regression by previous commit
Merge pull request #12 from r-e-d/master

Fix regression by previous commit