Fixes import error
Merge pull request #17 from digital-shokunin/patch-1

Fixes import error