Add missing import
Merge pull request #39 from pmav99/patch-1

Add missing import