From 4e3492b351cefe27637dc0f30e89c79f37cfd054 Mon Sep 17 00:00:00 2001
From: Yolan Uyttenhove <yolan.uyttenhove@ugent.be>
Date: Wed, 29 Nov 2023 16:49:03 +0100
Subject: [PATCH] make debug check stricter

---
 src/chemistry/EAGLE/chemistry.h | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/chemistry/EAGLE/chemistry.h b/src/chemistry/EAGLE/chemistry.h
index e1d8fcd1b6..52f92299f2 100644
--- a/src/chemistry/EAGLE/chemistry.h
+++ b/src/chemistry/EAGLE/chemistry.h
@@ -391,8 +391,12 @@ __attribute__((always_inline)) INLINE static void chemistry_kick_extra(
     for (int i = 0; i < chemistry_element_count; i++) {
       sum += p->chemistry_data.metal_mass_fraction[i];
     }
-    if (sum > 1.f)
-      error("Sum of element-wise metal mass fractions grew larger than 1!");
+    sum -= p->chemistry_data.metal_mass_fraction[chemistry_element_H];
+    sum -= p->chemistry_data.metal_mass_fraction[chemistry_element_He];
+    if (sum > p->chemistry_data.metal_mass_fraction_total)
+      error(
+          "Sum of element-wise metal mass fractions grew larger than total "
+          "metal mass fraction!");
 #endif
     chemistry_reset_mass_fluxes(p);
   }
-- 
GitLab