fixed missing import
Merge pull request #57 from fcurella/patch-4

fixed missing import