1 files changed, 6 insertions, 1 deletions
diff --git a/src/likelihood.c b/src/likelihood.c
index e045512..5f11f95 100644
--- a/src/likelihood.c
+++ b/src/likelihood.c
@@ -334,6 +334,7 @@ double nll_muon(event *ev, double T0, double *pos, double *dir, double t0, doubl
     double tmean = 0.0;
     muon_energy *m;
     double pmt_dir[3], R, cos_theta, theta, wavelength0, n_d2o, theta_cerenkov, s;
+    double logp_path;
 
     double mu_direct[MAX_PMTS];
     double ts[MAX_PMTS];
@@ -502,5 +503,9 @@ double nll_muon(event *ev, double T0, double *pos, double *dir, double t0, doubl
         }
     }
 
-    return kahan_sum(nll,nhit);
+    logp_path = 0.0;
+    for (i = 0; i < n; i++)
+        logp_path += pow(z1[i],2) + pow(z2[i],2);
+
+    return kahan_sum(nll,nhit) + logp_path;
 }