1 files changed, 5 insertions, 4 deletions
diff --git a/src/muon.c b/src/muon.c
index 52c73ed..0279984 100644
--- a/src/muon.c
+++ b/src/muon.c
@@ -240,16 +240,17 @@ double get_expected_charge(double x, double T, double *pos, double *dir, double
 
     R = NORM(pos);
 
+    /* FIXME: I just calculate delta assuming 400 nm light. */
+    wavelength0 = 400.0;
+
     if (R <= AV_RADIUS) {
         rho = HEAVY_WATER_DENSITY;
+        n = get_index_snoman_d2o(wavelength0);
     } else {
         rho = WATER_DENSITY;
+        n = get_index_snoman_h2o(wavelength0);
     }
 
-    /* FIXME: I just calculate delta assuming 400 nm light. */
-    wavelength0 = 400.0;
-    n = get_index(rho, wavelength0, 10.0);
-
     if (beta < 1/n) return 0;
 
     /* FIXME: is this formula valid for muons? */