diff --git a/rankers/modelling/__init__.py b/rankers/modelling/__init__.py
index 760103a..33b51c3 100644
--- a/rankers/modelling/__init__.py
+++ b/rankers/modelling/__init__.py
@@ -1,2 +1,2 @@
-from .cat import CatTransformer, Cat
+from .cat import CatTransformer, Cat, CatConfig
 from .dot import Dot, DotTransformer, DotConfig 
\ No newline at end of file