diff --git a/scripts/query_ai_model.py b/scripts/query_ai_model.py
index 40a546b..5a3424f 100644
--- a/scripts/query_ai_model.py
+++ b/scripts/query_ai_model.py
@@ -278,7 +278,6 @@ if __name__ == "__main__":
     experiment = mlflow.get_experiment_by_name("visual-semiotic-ai-analysis")
     experiment_id = experiment.experiment_id if experiment else None
     # Calculate accuracy
-    category_events = {}
     for category, initial_prompt in CATEGORY_PROMPT_MAP.items():
         step = 0
         run = mlflow.start_run(
@@ -309,8 +308,10 @@ if __name__ == "__main__":
                 continue
             try:
                 # Calculate accuracy for refined prompt
-                accuracy = calculate_model_prompt_accuracy(MODEL, refined_prompt, category)
-                mlflow.log_metric(f"accuracy", accuracy, step=step)
+                accuracy = calculate_model_prompt_accuracy(
+                    MODEL, refined_prompt, category
+                )
+                mlflow.log_metric("accuracy", accuracy, step=step)
             except Exception as e:
                 print(f"Error calculating accuracy: {e}")
                 continue