Add missing import
Merge pull request #7 from ksmandersen/master

Add missing import