diff --git a/warp/examples/sim/example_cloth.py b/warp/examples/sim/example_cloth.py
index 3d13ce34..5e38ba9e 100644
--- a/warp/examples/sim/example_cloth.py
+++ b/warp/examples/sim/example_cloth.py
@@ -100,7 +100,6 @@ def __init__(
                 tri_ka=1e4,
                 tri_kd=1e-5,
                 edge_ke=100,
-                color_particles=True,
             )
 
         usd_stage = Usd.Stage.Open(os.path.join(warp.examples.get_asset_directory(), "bunny.usd"))
@@ -122,6 +121,9 @@ def __init__(
             kf=1.0e1,
         )
 
+        if self.integrator_type == IntegratorType.VBD:
+            builder.color()
+
         self.model = builder.finalize()
         self.model.ground = True
         self.model.soft_contact_ke = 1.0e4