diff --git a/validator/main.py b/validator/main.py
index 5707c6b..b170f14 100644
--- a/validator/main.py
+++ b/validator/main.py
@@ -5,7 +5,7 @@
 
 # from guardrails.validator_base import ErrorSpan
 from guardrails.logger import logger
-from guardrails.validators import (
+from guardrails.validator_base import (
     FailResult,
     PassResult,
     ValidationResult,