diff --git a/examples/coupling/cloth_on_rigid.py b/examples/coupling/cloth_on_rigid.py
index cdbc347..f36dca0 100644
--- a/examples/coupling/cloth_on_rigid.py
+++ b/examples/coupling/cloth_on_rigid.py
@@ -12,7 +12,7 @@ def main():
     args = parser.parse_args()
 
     ########################## init ##########################
-    gs.init(seed=0, precision="32", logging_level="debug")
+    gs.init(seed=0, precision="32", logging_level="debug", backend=gs.cpu if args.cpu else gs.gpu)
 
     scene = gs.Scene(
         sim_options=gs.options.SimOptions(