diff --git a/test.py b/test.py
index 82178fc89338095efbf87c426eff6ad0052fbcbf..f2e10a3c71af04f498ec6b85ca03bea505dab28c 100644
--- a/test.py
+++ b/test.py
@@ -27,7 +27,7 @@ class RandomPredictor(AirbornePredictor):
     PARTICIPANT_TODO:
     During the evaluation all combinations for flight_id and flight_folder_path will be provided one by one.
 
-    NOTE: In case you want to load your model, please do so in `predict_setup` function.
+    NOTE: In case you want to load your model, please do so in `inference_setup` function.
     """
     def inference(self, flight_id):