diff --git a/samples/tuned_models.py b/samples/tuned_models.py
index 29246347d..d328d8c30 100644
--- a/samples/tuned_models.py
+++ b/samples/tuned_models.py
@@ -77,7 +77,7 @@ def test_tuned_models_generate_content(self):
         model = genai.GenerativeModel(model_name="tunedModels/my-increment-model")
         result = model.generate_content("III")
         print(result.text)  # "IV"
-        # [END tuned_models_create]
+        # [END tuned_models_generate_content]
 
     def test_tuned_models_get(self):
         # [START tuned_models_get]