diff --git a/test/torch_npu_schema.json b/test/torch_npu_schema.json
index c5c22598bd742317663ffc941b9e18c006a4b119..0fe3a425b9257de65f2a5820c912be34b9c94e56 100644
--- a/test/torch_npu_schema.json
+++ b/test/torch_npu_schema.json
@@ -789,7 +789,7 @@
     "signature": "(self)"
   },
   "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.npu_config.get_device_limit": {
     "signature": "(device)"