Add missing import
Merge pull request #175 from flibbertigibbet/feature/fix-import

Add missing import