Fixed import error in other module.
Merge pull request #3 from darkowlzz/importErr

Fixed import error in other module.