Fixed import error
Merge pull request #114 from kazoompa/master

Import error correction