Add some missing imports to controllers