Fix context error with new collective.transmogrifier