Add missing imports
Merge pull request #608 from JeroenDeDauw/missingimp

Add missing imports