fixed broken import
Merge pull request #25 from teharrison/master

fixed broken import