diff --git a/reformer/__init__.py b/reformer/__init__.py
index 0f8224e..1c557ba 100644
--- a/reformer/__init__.py
+++ b/reformer/__init__.py
@@ -3,7 +3,7 @@ from .lightning_module import TransformerLightningModule
 from .module import TransformerModel
 
 __all__ = [
-    "TransformerModel",
-    "TransformerLightningModule",
-    "TransformerEstimator",
+    "ReformerModel",
+    "ReformerLightningModule",
+    "ReformerEstimator",
 ]