fix import
Merge branch 'hotfix/v0.6.1'

* hotfix/v0.6.1:
  fix import