diff --git a/src/sp_validation/cosmo_val/core.py b/src/sp_validation/cosmo_val/core.py
index 066bf4a1..c6204c65 100644
--- a/src/sp_validation/cosmo_val/core.py
+++ b/src/sp_validation/cosmo_val/core.py
@@ -9,6 +9,7 @@
 import numpy as np
 import yaml
 from astropy.io import fits
+from cs_util.cosmo import get_cosmo
 from shear_psf_leakage import run_object, run_scale
 
 from ..b_modes import (