diff --git a/x2paddle/op_mapper/onnx2paddle/opset9/opset.py b/x2paddle/op_mapper/onnx2paddle/opset9/opset.py
index a151983bf7cded4297bb95ce3b5b3f0a63543d4f..ec4cd398e45c2aacaffcbcdc46c92877f58ecb58 100755
--- a/x2paddle/op_mapper/onnx2paddle/opset9/opset.py
+++ b/x2paddle/op_mapper/onnx2paddle/opset9/opset.py
@@ -59,15 +59,10 @@ def _rename_or_remove_weight(weights,
     Returns:
         None
     '''
-    # if origin_name not in weights:
-    #     raise KeyError('{} not a key in {}'.format(origin_name, weights.keys()))
-    # data = weights[origin_name]
-    # if target_name is not None:
-    #     # rename weight
-    #     weights[target_name] = data
     if origin_name not in weights:
         raise KeyError('{} not a key in {}'.format(origin_name, weights.keys()))
     if is_remove:
+        # TODO There might be a problem here.
         # remove weight
         data = weights.pop(origin_name)
     else: