Fix thrown NameError
Merge pull request #70 from eddiezab/patch-1

Fix thrown NameError, thanks eddiezab!