diff --git a/CybersecurityBenchmarks/benchmark/query_llm.py b/CybersecurityBenchmarks/benchmark/query_llm.py
index 8d423ab9..5dfa067a 100644
--- a/CybersecurityBenchmarks/benchmark/query_llm.py
+++ b/CybersecurityBenchmarks/benchmark/query_llm.py
@@ -36,7 +36,7 @@
     "speaking_language",
     "false_refusal_type",
     "caught_by_promptguard",
-    "category",
+    "categories",
     "subcategories",
     "voice_tone",
     "label",