diff --git a/src/hr_selection_function/__init__.py b/src/hr_selection_function/__init__.py
index 4144864..b272432 100644
--- a/src/hr_selection_function/__init__.py
+++ b/src/hr_selection_function/__init__.py
@@ -1,3 +1,3 @@
 from .data import set_data_directory
 from .models import HR24SelectionFunction, NStarsPredictor
-from .gaia import estimate_gaia_data_density, estimate_m10, estimate_m_rybizki
+from .gaia import M10Estimator, MSubsampleEstimator, GaiaDensityEstimator