diff --git a/aider/editors/__init__.py b/aider/editors/__init__.py
index e461a0e2c..ef7d0caea 100644
--- a/aider/editors/__init__.py
+++ b/aider/editors/__init__.py
@@ -1,2 +1,4 @@
+__all__ = ['EditBlockPrompts', 'WholeFilePrompts']
+
 from .editblock import EditBlockPrompts
 from .wholefile import WholeFilePrompts