diff --git a/.github/workflows/llmsecops-cicd.test.yml b/.github/workflows/llmsecops-cicd.test.yml
index 479c63b13..9b5d97cc4 100644
--- a/.github/workflows/llmsecops-cicd.test.yml
+++ b/.github/workflows/llmsecops-cicd.test.yml
@@ -30,5 +30,5 @@ jobs:
         pip install langchain_text_splitters
         pip install langchain_huggingface 
         pip install transformers
-        python -m tests.llm.llm
+        python -m tests.llm.llm --prompt "What is the capital of France?"
         
diff --git a/tests/llm/llm.py b/tests/llm/llm.py
index 57133ac36..adf6df160 100644
--- a/tests/llm/llm.py
+++ b/tests/llm/llm.py
@@ -1,7 +1,6 @@
 import argparse
 import os
-from langchain_core.output_parsers import StrOutputParser
-from langchain_core.runnables import RunnablePassthrough
+
 from langchain_community.document_loaders import WebBaseLoader
 from langchain_community.vectorstores import FAISS
 from langchain_text_splitters import RecursiveCharacterTextSplitter
@@ -37,10 +36,8 @@ class Llm:
 
     def get_response(self, input):
         # Use the model
-        print(input)
-        canned_input = "What is the capital of France?"
-        print(canned_input)
-        response = self.hf_model.invoke(canned_input)
+        print(f'End user prompt: {input}')
+        response = self.hf_model.invoke(input)
         print(response)
 
 if __name__ == "__main__":