diff --git a/nixos/tests/rxe.nix b/nixos/tests/rxe.nix
index 726dcc654048..194a2e3d2b94 100644
--- a/nixos/tests/rxe.nix
+++ b/nixos/tests/rxe.nix
@@ -32,18 +32,12 @@ in {
 
     client.wait_for_unit("default.target")
 
-    # ping pong test
-    server.succeed("screen -dmS rc_pingpong ibv_rc_pingpong -p 4800 -g0")
-    client.succeed("sleep 2; ibv_rc_pingpong -p 4800 -g0 server")
-
-    server.succeed("screen -dmS uc_pingpong ibv_uc_pingpong -p 4800 -g0")
-    client.succeed("sleep 2; ibv_uc_pingpong -p 4800 -g0 server")
-
-    server.succeed("screen -dmS ud_pingpong ibv_ud_pingpong -p 4800 -s 1024 -g0")
-    client.succeed("sleep 2; ibv_ud_pingpong -p 4800 -s 1024 -g0 server")
-
-    server.succeed("screen -dmS srq_pingpong ibv_srq_pingpong -p 4800 -g0")
-    client.succeed("sleep 2; ibv_srq_pingpong -p 4800 -g0 server")
+    # ping pong tests
+    for proto in "rc", "uc", "ud", "srq":
+        server.succeed(
+            "screen -dmS {0}_pingpong ibv_{0}_pingpong -p 4800 -s 1024 -g0".format(proto)
+        )
+        client.succeed("sleep 2; ibv_{}_pingpong -p 4800 -s 1024 -g0 server".format(proto))
 
     server.succeed("screen -dmS rping rping -s -a server -C 10")
     client.succeed("sleep 2; rping -c -a server -C 10")