fixed the import in an if statement error
Merge pull request #11 from ianrose/gh-10-fix-import-error

gh-10 fixed the import in an if statement error