Fix imports
Merge branch 'hotfix/0.1.2'