diff --git a/spacy/ml/callbacks.py b/spacy/ml/callbacks.py
index 9b24c71bd..393f208a6 100644
--- a/spacy/ml/callbacks.py
+++ b/spacy/ml/callbacks.py
@@ -23,7 +23,7 @@ DEFAULT_NVTX_ANNOTATABLE_PIPE_METHODS = [
     "update",
     "rehearse",
     "get_loss",
-    "get_student_teacher_loss",
+    "get_teacher_student_loss",
     "initialize",
     "begin_update",
     "finish_update",
diff --git a/website/docs/api/top-level.mdx b/website/docs/api/top-level.mdx
index 2eb3a84af..7e47d324a 100644
--- a/website/docs/api/top-level.mdx
+++ b/website/docs/api/top-level.mdx
@@ -921,7 +921,7 @@ backprop passes.
 Recursively wrap both the models and methods of each pipe using
 [NVTX](https://nvidia.github.io/NVTX/) range markers. By default, the following
 methods are wrapped: `pipe`, `predict`, `set_annotations`, `update`, `rehearse`,
-`get_loss`, `get_student_teacher_loss`, `initialize`, `begin_update`,
+`get_loss`, `get_teacher_student_loss`, `initialize`, `begin_update`,
 `finish_update`, `update`.
 
 | Name                        | Description                                                                                                                                                     |