diff --git a/src/components/BpmnDesign/assets/module/Palette/CustomPaletteProvider.ts b/src/components/BpmnDesign/assets/module/Palette/CustomPaletteProvider.ts
index e011fb3..fe8fd78 100644
--- a/src/components/BpmnDesign/assets/module/Palette/CustomPaletteProvider.ts
+++ b/src/components/BpmnDesign/assets/module/Palette/CustomPaletteProvider.ts
@@ -80,7 +80,7 @@ class CustomPaletteProvider extends PaletteProvider {
 
     // 定义一个函数来创建自定义用户任务
     function createCustomUserTask(event) {
-      const bpmnFactory: BpmnFactory | undefined = modeler().getModeler().get('bpmnFactory');
+      const bpmnFactory: BpmnFactory | undefined = modeler().getBpmnFactory();
       const customUserTask = elementFactory.createShape({
         type: 'bpmn:UserTask',
         businessObject: bpmnFactory.create('bpmn:UserTask', {
@@ -92,7 +92,7 @@ class CustomPaletteProvider extends PaletteProvider {
     }
 
     function createMultiInstanceUserTask(event) {
-      const bpmnFactory: BpmnFactory | undefined = modeler().getModeler().get('bpmnFactory');
+      const bpmnFactory: BpmnFactory | undefined = modeler().getBpmnFactory();
       // 创建一个 bpmn:UserTask
       const userTask = bpmnFactory.create('bpmn:UserTask', {
         // name: '多实例用户任务', // 在画板中显示字段