diff --git a/test/torch_npu_schema.json b/test/torch_npu_schema.json
index 054d32329a2fc85029e8c271333ebd3570fd00f6..dd868b13b32f5df859ab0421fcb1ea11ae02024d 100644
--- a/test/torch_npu_schema.json
+++ b/test/torch_npu_schema.json
@@ -846,7 +846,7 @@
     "signature": "(*args, **kwargs)"
   },
   "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.profiler.HostSystem": {
     "signature": "()"