bug fix: broken import
Merge pull request #39 from VasylZarva/master

bug fix: broken import