diff --git a/java/client/src/test/java/glide/api/models/TransactionTests.java b/java/client/src/test/java/glide/api/models/TransactionTests.java
index 04fb36d531..4c30e60e06 100644
--- a/java/client/src/test/java/glide/api/models/TransactionTests.java
+++ b/java/client/src/test/java/glide/api/models/TransactionTests.java
@@ -384,12 +384,8 @@ public void transaction_builds_protobuf_request(BaseTransaction<?> transaction)
         transaction.zpopmin("key", 2);
         results.add(Pair.of(ZPopMin, buildArgs("key", "2")));
 
-        // TODO use buildArgs after merge/rebase on main
         transaction.bzpopmin(new String[] {"key1", "key2"}, .5);
-        results.add(
-                Pair.of(
-                        BZPopMin,
-                        ArgsArray.newBuilder().addArgs("key1").addArgs("key2").addArgs("0.5").build()));
+        results.add(Pair.of(BZPopMin, buildArgs("key1", "key2", "0.5")));
 
         transaction.zpopmax("key");
         results.add(Pair.of(ZPopMax, buildArgs("key")));