diff --git a/csiborgtools/flow/__init__.py b/csiborgtools/flow/__init__.py
index 6ee3f87..4d8a2c5 100644
--- a/csiborgtools/flow/__init__.py
+++ b/csiborgtools/flow/__init__.py
@@ -18,4 +18,5 @@ from .flow_model import (PV_LogLikelihood, PV_validation_model, dist2redshift,
                          Observed2CosmologicalRedshift, predict_zobs,           # noqa
                          project_Vext, stack_pzosmo_over_realizations)          # noqa
 from .selection import ToyMagnitudeSelection                                    # noqa
-from .void_model import load_void_data, interpolate_void                        # noqa
+from .void_model import (load_void_data, interpolate_void, select_void_h,       # noqa
+                         mock_void)                                             # noqa