diff --git a/src/gretel_trainer/relational/__init__.py b/src/gretel_trainer/relational/__init__.py
index 01532e12..ff9c215f 100644
--- a/src/gretel_trainer/relational/__init__.py
+++ b/src/gretel_trainer/relational/__init__.py
@@ -8,5 +8,6 @@
     sqlite_conn,
 )
 from gretel_trainer.relational.core import RelationalData
+from gretel_trainer.relational.extractor import ExtractorConfig
 from gretel_trainer.relational.log import set_log_level
 from gretel_trainer.relational.multi_table import MultiTable