fixed missing import
Merge pull request #28 from weseek/master

fixed missing import