diff --git a/src/chemistry/gear/chemistry.h b/src/chemistry/gear/chemistry.h
index 5dc7a48939cdb298233abbe358473f59fc7e72f4..e9d3d00febe2438c52728f94d0136fb72ff8dc3b 100644
--- a/src/chemistry/gear/chemistry.h
+++ b/src/chemistry/gear/chemistry.h
@@ -103,9 +103,11 @@ __attribute__((always_inline)) INLINE static void chemistry_init_part(
  *
  * @param p The particle to act upon.
  * @param cd #chemistry_data containing chemistry informations.
+ * @param cosmo The current cosmological model.
  */
 __attribute__((always_inline)) INLINE static void chemistry_end_density(
-    struct part* restrict p, const struct chemistry_data* cd) {
+    struct part* restrict p, const struct chemistry_data* cd,
+    const struct cosmology* cosmo) {
 
   /* Some smoothing length multiples. */
   const float h = p->h;
diff --git a/src/runner.c b/src/runner.c
index e32d20a208cd98a94cf303cf4253e49426006f58..fb74b20564482cbf88ddcd07fd99f325f93d3414 100644
--- a/src/runner.c
+++ b/src/runner.c
@@ -756,7 +756,7 @@ void runner_do_ghost(struct runner *r, struct cell *c, int timer) {
 
             /* Re-initialise everything */
             hydro_init_part(p, hs);
-            chemistry_init_part(p, e->chemistry);
+            chemistry_init_part(p, chemistry);
 
             /* Off we go ! */
             continue;