diff --git a/test/torch_npu_schema.json b/test/torch_npu_schema.json
index 388bcaf896ffd58eca56362f7fd94ef6ad2d845c..7e6f89bcd913b471bf49345582a2127aa891350e 100644
--- a/test/torch_npu_schema.json
+++ b/test/torch_npu_schema.json
@@ -2526,7 +2526,7 @@
     "signature": "(obj: object, f, pickle_module: Any = <module 'pickle'>, pickle_protocol: int = 2, _use_new_zipfile_serialization: bool = True, _disable_byteorder_record: bool = False, model: torch.nn.modules.module.Module = None) -> None"
   },
   "torch_npu.dynamo.torchair.ops.npu_record_tagged_stream": {
-    "signature": "(self: torch.Tensor, tagged_stream: str)"
+    "signature": "(input: torch.Tensor, tagged_stream: str)"
   },
   "torch_npu.npu_bounding_box_encode": {
     "signature": "(anchor_box, ground_truth_box, means0, means1, means2, means3, stds0, stds1, stds2, stds3)"