Add missing imports to visitors