diff --git a/pm4py/algo/querying/llm/connectors/openai.py b/pm4py/algo/querying/llm/connectors/openai.py
index 1fba40431..2ad5f1119 100644
--- a/pm4py/algo/querying/llm/connectors/openai.py
+++ b/pm4py/algo/querying/llm/connectors/openai.py
@@ -24,4 +24,5 @@ def apply(query: str, parameters: Optional[Dict[Any, Any]] = None) -> str:
     message = {"role": "user", "content": query}
 
     response = client.chat.completions.create(model=model, messages=[message])
-    return response["choices"][0]["message"]["content"]
+
+    return response.choices[0].message.content