diff --git a/src/testdatagen/generators/__init__.py b/src/testdatagen/generators/__init__.py
index b203bae..c3c3f4f 100644
--- a/src/testdatagen/generators/__init__.py
+++ b/src/testdatagen/generators/__init__.py
@@ -1 +1,5 @@
 """Generators package for TestDataGen."""
+
+from .json_schema_generator import JSONSchemaGenerator
+
+__all__ = ["JSONSchemaGenerator"]