diff --git a/csiborgtools/flow/__init__.py b/csiborgtools/flow/__init__.py
index 9061201..2fb6057 100644
--- a/csiborgtools/flow/__init__.py
+++ b/csiborgtools/flow/__init__.py
@@ -12,9 +12,9 @@
 # You should have received a copy of the GNU General Public License along
 # with this program; if not, write to the Free Software Foundation, Inc.,
 # 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
-from .flow_model import (DataLoader, PV_LogLikelihood, PV_validation_model,     # noqa
-                         dist2redshift, get_model,                              # noqa
+from .io import (DataLoader, get_model, read_absolute_calibration,              # noqa
+                 radial_velocity_los)                                           # noqa
+from .flow_model import (PV_LogLikelihood, PV_validation_model, dist2redshift,  # noqa
                          Observed2CosmologicalRedshift, predict_zobs,           # noqa
-                         project_Vext, radial_velocity_los,                     # noqa
-                         stack_pzosmo_over_realizations)                        # noqa
+                         project_Vext, stack_pzosmo_over_realizations)          # noqa
 from .selection import ToyMagnitudeSelection                                    # noqa