diff --git a/test/torch_npu_schema.json b/test/torch_npu_schema.json
index be0fd8d3edc8cb1732c00625dffcc69b842925e4..115615178adca6ae0485bf617ea85fc7adc4c9db 100644
--- a/test/torch_npu_schema.json
+++ b/test/torch_npu_schema.json
@@ -2697,7 +2697,7 @@
     "signature": "(self: torch.Tensor, dependency: torch.Tensor)"
   },
    "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.dynamo.torchair.ops.npu_create_tagged_event": {
     "signature": "(tag: str)"