Add missing import
Merge pull request #20 from ionelmc/patch-1

Add missing import