diff --git a/integration-tests/gaudi/test_gaudi_generate.py b/integration-tests/gaudi/test_gaudi_generate.py
index d2adf2b2..4ed37dad 100644
--- a/integration-tests/gaudi/test_gaudi_generate.py
+++ b/integration-tests/gaudi/test_gaudi_generate.py
@@ -3,7 +3,7 @@ from typing import Any, Dict
 from text_generation import AsyncClient
 import pytest
 
-# The "args" config is not optimized for speed but only check that the inference is working for the different models architectures
+# The "args" values in TEST_CONFIGS are not optimized for speed but only check that the inference is working for the different models architectures.
 TEST_CONFIGS = {
     "meta-llama/Llama-3.1-8B-Instruct-shared": {
         "model_id": "meta-llama/Llama-3.1-8B-Instruct",