diff --git a/tests/guardrails/test_prompt_injection_llm_guardrail.py b/tests/guardrails/test_prompt_injection_llm_guardrail.py
index 24f1a725fab7..a4f604fc4321 100644
--- a/tests/guardrails/test_prompt_injection_llm_guardrail.py
+++ b/tests/guardrails/test_prompt_injection_llm_guardrail.py
@@ -2,7 +2,10 @@
 from openai import OpenAI
 
 from weave.guardrails import PromptInjectionLLMGuardrail
-from weave.guardrails.prompt_injection_guardrail import LLMGuardrailResponse, LLMGuardrailReasoning
+from weave.guardrails.prompt_injection_guardrail import (
+    LLMGuardrailReasoning,
+    LLMGuardrailResponse,
+)
 
 
 # mock the create function