diff --git a/test/partitions_SUITE.erl b/test/partitions_SUITE.erl
index ff8d5f5f134a..7425bec30769 100644
--- a/test/partitions_SUITE.erl
+++ b/test/partitions_SUITE.erl
@@ -389,12 +389,10 @@ partitions(Node) ->
     end.
 
 block(X, Y) ->
-    rpc:call(X, inet_tcp_proxy, block, [Y]),
-    rpc:call(Y, inet_tcp_proxy, block, [X]).
+    rabbit_ct_broker_helpers:block_traffic_between(X, Y).
 
 allow(X, Y) ->
-    rpc:call(X, inet_tcp_proxy, allow, [Y]),
-    rpc:call(Y, inet_tcp_proxy, allow, [X]).
+    rabbit_ct_broker_helpers:allow_traffic_between(X, Y).
 
 await_running   (Node, Bool)  -> await(Node, Bool,  fun is_running/1).
 await_listening (Node, Bool)  -> await(Node, Bool,  fun is_listening/1).