diff --git a/tests/guardrails_tests/test_aim_guardrail_base_url.py b/tests/guardrails_tests/test_aim_guardrail_base_url.py
index 105ddc89b17..3adb40db2c3 100644
--- a/tests/guardrails_tests/test_aim_guardrail_base_url.py
+++ b/tests/guardrails_tests/test_aim_guardrail_base_url.py
@@ -1,4 +1,4 @@
-import pytest
+from litellm.proxy.guardrails.guardrail_hooks.aim.aim import AimGuardrail
 from litellm.proxy.guardrails.guardrail_hooks.aim.aim import AimGuardrail
 import os