add missing imports
Merge pull request #1 from marlier/1.7-fix-import

add missing imports