diff --git a/transformer/__init__.py b/transformer/__init__.py
index 6f95d1d..42e203e 100644
--- a/transformer/__init__.py
+++ b/transformer/__init__.py
@@ -1,9 +1,9 @@
 from .module import TransformerModel
 from .lightning_module import TransformerLightningModule
-from .estimator import TransformeEstimator
+from .estimator import TransformerEstimator
 
 __all__ = [
     "TransformerModel",
     "TransformerLightningModule",
-    "TransformeEstimator",
+    "TransformerEstimator",
 ]