diff --git a/runtime/vtk.hpp b/runtime/vtk.hpp
index 2fed101ce630eef636610ac54c0489ea42177e81..f122565694634594e0733017df8624a280681fc0 100644
--- a/runtime/vtk.hpp
+++ b/runtime/vtk.hpp
@@ -31,6 +31,7 @@ void vtk_write_data(PairsSimulation *ps, const char *filename, int start, int en
 
     ps->copyPropertyToHost(masses, ReadOnly);
     ps->copyPropertyToHost(positions, ReadOnly);
+    ps->copyPropertyToHost(flags, ReadOnly);
 
     for(int i = start; i < end; i++) {
         if(flags(i) & FLAGS_INFINITE) {
diff --git a/src/pairs/sim/comm.py b/src/pairs/sim/comm.py
index 9462228203d3a8cf5c03aad0381821bdb4232355..3a439d8048517f2b4c8b0cb5023c1942787d71bd 100644
--- a/src/pairs/sim/comm.py
+++ b/src/pairs/sim/comm.py
@@ -47,7 +47,7 @@ class Comm:
         # Every property that is not constant across timesteps and have neighbor accesses during any
         # interaction kernel (i.e. property[j] in force calculation kernel)
         prop_names = ['position', 'linear_velocity', 'angular_velocity']
-        prop_list = [ self.sim.property(p) for p in prop_names if self.sim.property(p) is not None]
+        prop_list = [self.sim.property(p) for p in prop_names if self.sim.property(p) is not None]
 
         PackAllGhostParticles(self, prop_list)
         CommunicateAllData(self, prop_list)