Fix: Add missing import
Merge pull request #7 from localheinz/bugfix/import

Fix: Add missing import