diff --git a/tupak/likelihood.py b/tupak/likelihood.py
index 47e70944d59a2220f19ddf1fcb93ce6322fd21bf..c2a218e3004c4033b5e8d0de72b655a01a818520 100644
--- a/tupak/likelihood.py
+++ b/tupak/likelihood.py
@@ -73,12 +73,23 @@ class GravitationalWaveTransient(Likelihood):
             self.delta_distance = 0
             self.distance_prior_array = np.array([])
             self.setup_distance_marginalization()
-            prior['luminosity_distance'] = 1
+            self.prior['luminosity_distance'] = 1
 
         if self.phase_marginalization:
             self.bessel_function_interped = None
             self.setup_phase_marginalization()
-            prior['psi'] = 0
+            self.prior['psi'] = 0
+
+    @property
+    def prior(self):
+        return self.__prior
+
+    @prior.setter
+    def prior(self, prior):
+        if prior is not None:
+            self.__prior = prior
+        else:
+            self.__prior = dict()
 
     def noise_log_likelihood(self):
         log_l = 0