Fix name in import

Fixes #1