diff --git a/src/swiftest/swiftest_util.f90 b/src/swiftest/swiftest_util.f90
index 7290b4b29..d7696abc2 100644
--- a/src/swiftest/swiftest_util.f90
+++ b/src/swiftest/swiftest_util.f90
@@ -2539,7 +2539,7 @@ module subroutine swiftest_util_setup_body(self, n, param)
       self%peri(:)   = 0.0_DP
       self%atp(:)    = 0.0_DP
 
-      if (param%loblatecb) then
+      if (param%loblatecb .or. param%lshgrav) then
          allocate(self%aobl(NDIM, n))
          self%aobl(:,:) = 0.0_DP
       end if