diff --git a/examples/petsc/dmswarm.c b/examples/petsc/dmswarm.c
index 1dc3618fd8..113730e15a 100644
--- a/examples/petsc/dmswarm.c
+++ b/examples/petsc/dmswarm.c
@@ -398,7 +398,7 @@ PetscErrorCode DMSwarmInterpolateFromCellToSwarm_Petsc(DM dm_swarm, const char *
     PetscCall(DMRestoreWorkArray(dm_mesh, num_points_in_cell * dim, MPIU_REAL, &coords_points_cell_true));
     PetscCall(DMRestoreWorkArray(dm_mesh, num_points_in_cell * dim, MPIU_REAL, &coords_points_cell_ref));
     PetscCall(PetscTabulationDestroy(&tabulation));
-    PetscCall(PetscFree(points_cell));
+    PetscCall(DMSwarmSortRestorePointsPerCell(dm_swarm, cell, &num_points_in_cell, &points_cell));
   }
 
   // Cleanup
@@ -486,7 +486,7 @@ PetscErrorCode DMSwarmCheckSwarmValues(DM dm_swarm, const char *field, PetscScal
     }
 
     // -- Cleanup
-    PetscCall(PetscFree(points));
+    PetscCall(DMSwarmSortRestorePointsPerCell(dm_swarm, cell, &num_points_in_cell, &points));
   }
 
   // Cleanup