fix preset models test

dangilman · Sep 5, 2024 · e7e42fe · e7e42fe
1 parent 3214bb2
commit e7e42fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyHalo/PresetModels/wdm.py b/pyHalo/PresetModels/wdm.py
@@ -106,7 +106,7 @@ def WDM(z_lens, z_source, log_mc, sigma_sub=0.025, log_mlow=6., log_mhigh=10., l
     kwargs_mc_field['concentration_cdm_class'] = concentration_model_CDM
     concentration_model_fieldhalos = model_fieldhalos(**kwargs_mc_field)
     if c_host is None:
-        c_host = concentration_model_CDM.nfw_concentration(10 ** log_m_host, z_lens)
+        c_host = concentration_model_fieldhalos.nfw_concentration(10 ** log_m_host, z_lens)
 
     # SET THE TRUNCATION RADIUS FOR SUBHALOS AND FIELD HALOS
     kwargs_truncation_model_subhalos['lens_cosmo'] = pyhalo.lens_cosmo