diff --git a/src/gravity/MultiSoftening/gravity.h b/src/gravity/MultiSoftening/gravity.h
index 935f47c42045dd925f2ca6eddb2d8fbfcecf8830..08282a038dc67c2740e3d5e6ba95a5a5b8a190ff 100644
--- a/src/gravity/MultiSoftening/gravity.h
+++ b/src/gravity/MultiSoftening/gravity.h
@@ -62,11 +62,6 @@ __attribute__((always_inline)) INLINE static float gravity_get_softening(
     default:
       return 0.f;
   }
-
-  /* if (gp->type == swift_type_dark_matter_background) */
-  /*   return grav_props->epsilon_background_fac * cbrtf(gp->mass); */
-  /* else */
-  /*   return grav_props->epsilon_cur; */
 }
 
 /**
diff --git a/src/gravity/Potential/gravity.h b/src/gravity/Potential/gravity.h
index 2e79af643bea45144397f45cab551b22c408efa4..3a3e9680016ef51c554691c6f43f8279cde8caf9 100644
--- a/src/gravity/Potential/gravity.h
+++ b/src/gravity/Potential/gravity.h
@@ -22,6 +22,7 @@
 
 #include <float.h>
 
+/* Local includes. */
 #include "cosmology.h"
 #include "gravity_properties.h"
 #include "kernel_gravity.h"
diff --git a/tests/testGravityDerivatives.c b/tests/testGravityDerivatives.c
index f31967de7075bccfb2c7fb19c1ba262aa12da54f..032460d9f519c455e485e0dddfc5f8742138d60a 100644
--- a/tests/testGravityDerivatives.c
+++ b/tests/testGravityDerivatives.c
@@ -955,12 +955,11 @@ int main(int argc, char* argv[]) {
     const double r_inv = 1. / sqrt(r2);
     const double r = r2 * r_inv;
     const double eps = r / 10.;
-    const double eps_inv = 1. / eps;
 
     /* Compute all derivatives */
     struct potential_derivatives_M2L pot;
-    potential_derivatives_compute_M2L(dx, dy, dz, r2, r_inv, eps, eps_inv,
-                                      periodic, r_s_inv, &pot);
+    potential_derivatives_compute_M2L(dx, dy, dz, r2, r_inv, eps, periodic,
+                                      r_s_inv, &pot);
 
     /* Minimal value we care about */
     const double min = 1e-9;
diff --git a/tests/testPotentialPair.c b/tests/testPotentialPair.c
index 064c86d42f8df907d1ffaaab164b6a2f8b534b19..d5fbda36a9ef79352f79627b5cef908030401da2 100644
--- a/tests/testPotentialPair.c
+++ b/tests/testPotentialPair.c
@@ -125,7 +125,8 @@ int main(int argc, char *argv[]) {
 
   struct gravity_props props;
   props.theta_crit2 = 0.;
-  props.epsilon_cur = eps;
+  props.epsilon_DM_cur = eps;
+  props.epsilon_baryon_cur = eps;
   e.gravity_properties = &props;
 
   struct runner r;
@@ -386,7 +387,7 @@ int main(int argc, char *argv[]) {
 
   /* Now let's make a multipole out of it. */
   gravity_reset(ci.grav.multipole);
-  gravity_P2M(ci.grav.multipole, ci.grav.parts, ci.grav.count);
+  gravity_P2M(ci.grav.multipole, ci.grav.parts, ci.grav.count, &props);
 
   gravity_multipole_print(&ci.grav.multipole->m_pole);
 
diff --git a/tests/testPotentialSelf.c b/tests/testPotentialSelf.c
index 10eb499570a591daaf0de2e011f2346077905e8e..ff55558aff77441458e822cfa11922bd2dabfa06 100644
--- a/tests/testPotentialSelf.c
+++ b/tests/testPotentialSelf.c
@@ -115,7 +115,8 @@ int main(int argc, char *argv[]) {
 
   struct gravity_props props;
   props.a_smooth = 1.25;
-  props.epsilon_cur = eps;
+  props.epsilon_DM_cur = eps;
+  props.epsilon_baryon_cur = eps;
   e.gravity_properties = &props;
 
   struct runner r;
diff --git a/tests/testReading.c b/tests/testReading.c
index b2cf743a066920c3d28ea8768334a6a8b1c9b5f0..47f8d0dfdc043985c6344abecf31e605226e6899 100644
--- a/tests/testReading.c
+++ b/tests/testReading.c
@@ -28,7 +28,7 @@
 
 int main(int argc, char *argv[]) {
 
-  size_t Ngas = 0, Ngpart = 0, Nspart = 0, Nbpart = 0;
+  size_t Ngas = 0, Ngpart = 0, Ngpart_background = 0, Nspart = 0, Nbpart = 0;
   int flag_entropy_ICs = -1;
   int i, j, k;
   double dim[3];
@@ -51,8 +51,15 @@ int main(int argc, char *argv[]) {
 
   /* Read data */
   read_ic_single("input.hdf5", &us, dim, &parts, &gparts, &sparts, &bparts,
-                 &Ngas, &Ngpart, &Nspart, &Nbpart, &flag_entropy_ICs, 1, 1, 0,
-                 0, 0, 0, 1., 1., 1, 0);
+                 &Ngas, &Ngpart, &Ngpart_background, &Nspart, &Nbpart,
+                 &flag_entropy_ICs,
+                 /*with_hydro=*/1,
+                 /*with_gravity=*/1,
+                 /*with_stars=*/0,
+                 /*with_black_holes=*/0,
+                 /*cleanup_h=*/0,
+                 /*cleanup_sqrt_a=*/0,
+                 /*h=*/1., /*a=*/1., /*n_threads=*/1, /*dry_run=*/0);
 
   /* Check global properties read are correct */
   assert(dim[0] == boxSize);
diff --git a/tests/testSelectOutput.c b/tests/testSelectOutput.c
index 0be250e48cb2748093c8cd8c1381303c3060c9ba..9797406d91b74c7e9effe14da7b4f0100d901148 100644
--- a/tests/testSelectOutput.c
+++ b/tests/testSelectOutput.c
@@ -84,7 +84,7 @@ int main(int argc, char *argv[]) {
   clocks_set_cpufreq(cpufreq);
 
   char *base_name = "testSelectOutput";
-  size_t Ngas = 0, Ngpart = 0, Nspart = 0, Nbpart = 0;
+  size_t Ngas = 0, Ngpart = 0, Ngpart_background = 0, Nspart = 0, Nbpart = 0;
   int flag_entropy_ICs = -1;
   int periodic = 1;
   double dim[3];
@@ -112,8 +112,15 @@ int main(int argc, char *argv[]) {
   /* Read data */
   message("Reading initial conditions.");
   read_ic_single("input.hdf5", &us, dim, &parts, &gparts, &sparts, &bparts,
-                 &Ngas, &Ngpart, &Nspart, &Nbpart, &flag_entropy_ICs, 1, 0, 0,
-                 0, 0, 0, 1., 1., 1, 0);
+                 &Ngas, &Ngpart, &Ngpart_background, &Nspart, &Nbpart,
+                 &flag_entropy_ICs,
+                 /*with_hydro=*/1,
+                 /*with_gravity=*/0,
+                 /*with_stars=*/0,
+                 /*with_black_holes=*/0,
+                 /*cleanup_h=*/0,
+                 /*cleanup_sqrt_a=*/0,
+                 /*h=*/1., /*a=*/1., /*n_threads=*/1, /*dry_run=*/0);
 
   /* pseudo initialization of the space */
   message("Initialization of the space.");