diff --git a/agentic_security/probe_data/test_data.py b/agentic_security/probe_data/test_data.py
index 6e391c8..a4398f6 100644
--- a/agentic_security/probe_data/test_data.py
+++ b/agentic_security/probe_data/test_data.py
@@ -12,13 +12,13 @@ def test_empty_dataset_list(self):
         # Assert that the prepared_prompts list is empty
         assert prepared_prompts == []
 
-        assert len(
-            prepare_prompts(["markush1/LLM-Jailbreak-Classifier"], 100)
-        ) == snapshot(1)
+        # assert len(
+        #     prepare_prompts(["markush1/LLM-Jailbreak-Classifier"], 100)
+        # ) == snapshot(1)
 
         assert len(
             prepare_prompts(
-                ["markush1/LLM-Jailbreak-Classifier", "llm-adaptive-attacks"],
+                ["llm-adaptive-attacks"],
                 100,
             )
-        ) == snapshot(2)
+        ) == snapshot(1)