diff --git a/src/engine.c b/src/engine.c
index e782fc0ed09bf05c994e53a3f47a5a74224fe28d..12047e81c7c0f20eaa3e6c388a0743a676e3a81d 100644
--- a/src/engine.c
+++ b/src/engine.c
@@ -2621,8 +2621,8 @@ void engine_prepare(struct engine *e) {
      * That can include cells that have not
      * previously been active on this rank. */
     space_check_drift_point(e->s, e->ti_old);
-#endif
   }
+#endif
 
   /* Do we need repartitioning ? */
   if (e->forcerepart) engine_repartition(e);