Fixing deprecated module import
Merge pull request #37 from bellhops/master

Fixing deprecated module import