diff --git a/test/torch_npu_schema.json b/test/torch_npu_schema.json
index e6854328145528e0d019264436b44d5daff69b0b..75637cc98766d9417600b10404649b36017fcfa9 100644
--- a/test/torch_npu_schema.json
+++ b/test/torch_npu_schema.json
@@ -777,7 +777,7 @@
     "signature": "(self, cpu_update_input)"
   },
   "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.graphs.NPUGraph": {
     "signature": "()"