diff --git a/validator/__init__.py b/validator/__init__.py
index 72a2623..fe4bb95 100644
--- a/validator/__init__.py
+++ b/validator/__init__.py
@@ -1,3 +1,3 @@
-from .main import ValidatorTemplate
+from .main import LlamaGuard7B
 
-__all__ = ["ValidatorTemplate"]
+__all__ = ["LlamaGuard7B"]