diff --git a/x2paddle/core/program.py b/x2paddle/core/program.py
index 0cc78c5b8caa4dcb08345a05b24364c60f3ec214..31e0e44b5c169b27933dc8e6bceac67c191a0771 100644
--- a/x2paddle/core/program.py
+++ b/x2paddle/core/program.py
@@ -322,7 +322,7 @@ class PaddleGraph(object):
         if self.source_type == "caffe":
             custom_import = "from x2paddle.op_mapper.static.caffe2paddle " + \
                              "import caffe_custom_layer as x2paddle_nn"
-        if self.source_type == "onnx":
+        elif self.source_type == "onnx":
             custom_import = "from x2paddle.op_mapper.static.onnx2paddle " + \
                              "import onnx_custom_layer as x2paddle_nn"
         else: