Fix regression
Merge pull request #46 from id774/master

Fix regression