diff --git a/tests/guardrails_tests/test_ztds_guardrail.py b/tests/guardrails_tests/test_ztds_guardrail.py
index be4b6a85da1..2db0507b265 100644
--- a/tests/guardrails_tests/test_ztds_guardrail.py
+++ b/tests/guardrails_tests/test_ztds_guardrail.py
@@ -6,7 +6,12 @@ https://datatracker.ietf.org/doc/draft-sibiryakov-ztds-protocol/
 
 import asyncio
 import unittest
-from litellm.proxy.guardrails.guardrail_hooks.ztds import ZTDSGuardrail
+import os, sys
+try:
+    from litellm.proxy.guardrails.guardrail_hooks.ztds import ZTDSGuardrail
+except ImportError:
+    sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+    from ztds import ZTDSGuardrail
 
 
 class MockMessage: