Add missing imports

These imports seems to be required