Fixed import issue

Fixes #68