[IMP] Adapt module to be compatible with IrModelExport.force_execute_actions