fixed missing import
Merge pull request #1 from mbruggerbp/master

fixed missing import