Fixed import
Merge pull request #11 from mdsol/feature/doctweak

Fixed import - LGTM, merging