Fix regression.
Merge pull request #326 from tobeytailor/master

Fix regression.