Fix local import
Merge pull request #16 from hattya/master

Fix local import