diff --git a/test/test_Likelihood/test_cosmo_likelihood.py b/test/test_Likelihood/test_cosmo_likelihood.py
index dc32373..6035cba 100644
--- a/test/test_Likelihood/test_cosmo_likelihood.py
+++ b/test/test_Likelihood/test_cosmo_likelihood.py
@@ -78,7 +78,7 @@ def test_log_likelihood(self):
             cosmo_fixed=None,
         )
 
-        def custom_prior(kwargs_cosmo, kwargs_lens, kwargs_kin, kwargs_source):
+        def custom_prior(kwargs_cosmo, kwargs_lens, kwargs_kin, kwargs_source, kwargs_los):
             return -1
 
         cosmoL_prior = CosmoLikelihood(