fixing annotate

FlamTeam · Jan 23, 2024 · 26f09e2 · 26f09e2
1 parent e0c9a9e
commit 26f09e2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flamedisx/lxe_blocks/raw_signals.py b/flamedisx/lxe_blocks/raw_signals.py
@@ -39,7 +39,7 @@ def _simulate(self, d):
     def _annotate(self, d):
         m = self.gimme_numpy(self.quanta_name + '_gain_mean')
         if self.quanta_name == 'electron':
-            m *= self.gimme_numpy('geometrical_acceptance',d['s2_raw'].values/m)
+            m *= self.gimme_numpy('geometrical_acceptance',d['s2_raw_mle'].values/m)
         s = self.gimme_numpy(self.quanta_name + '_gain_std')
 
         mle = d[self.quanta_name + 's_detected_mle'] = \