diff --git a/src/jinjax/__init__.py b/src/jinjax/__init__.py
index e30f39b..0c61e8d 100644
--- a/src/jinjax/__init__.py
+++ b/src/jinjax/__init__.py
@@ -1,5 +1,18 @@
-from .catalog import Catalog  # noqa
-from .component import Component  # noqa
+from .catalog import Catalog
+from .component import Component
 from .exceptions import *  # noqa
-from .jinjax import JinjaX  # noqa
-from .html_attrs import HTMLAttrs, LazyString  # noqa
+from .html_attrs import HTMLAttrs, LazyString
+from .jinjax import JinjaX
+
+
+__all__ = [
+    "Catalog",
+    "Component",
+    "ComponentNotFound",
+    "DuplicateDefDeclaration",
+    "HTMLAttrs",
+    "InvalidArgument",
+    "JinjaX",
+    "LazyString",
+    "MissingRequiredArgument",
+]