Fix ImportError
Merge pull request #116 from iskra/patch-1

Fix ImportError