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

bug fix: broken import