diff --git a/apitest/src/test/java/bisq/apitest/method/wallet/BsqWalletTest.java b/apitest/src/test/java/bisq/apitest/method/wallet/BsqWalletTest.java
index a9bf4ca09e3..861c1839c5b 100644
--- a/apitest/src/test/java/bisq/apitest/method/wallet/BsqWalletTest.java
+++ b/apitest/src/test/java/bisq/apitest/method/wallet/BsqWalletTest.java
@@ -178,7 +178,7 @@ public static void tearDown() {
     }
 
     private void verifyBsqBalances(bisq.core.api.model.BsqBalanceInfo expected,
-                                   bisq.proto.grpc.BsqBalanceInfo actual) {
+                                   BsqBalanceInfo actual) {
         assertEquals(expected.getAvailableConfirmedBalance(), actual.getAvailableConfirmedBalance());
         assertEquals(expected.getUnverifiedBalance(), actual.getUnverifiedBalance());
         assertEquals(expected.getUnconfirmedChangeBalance(), actual.getUnconfirmedChangeBalance());
diff --git a/apitest/src/test/java/bisq/apitest/method/wallet/BtcWalletTest.java b/apitest/src/test/java/bisq/apitest/method/wallet/BtcWalletTest.java
index fe0949f9425..380198d53ee 100644
--- a/apitest/src/test/java/bisq/apitest/method/wallet/BtcWalletTest.java
+++ b/apitest/src/test/java/bisq/apitest/method/wallet/BtcWalletTest.java
@@ -93,7 +93,7 @@ public static void tearDown() {
     }
 
     private void verifyBtcBalances(bisq.core.api.model.BtcBalanceInfo expected,
-                                   bisq.proto.grpc.BtcBalanceInfo actual) {
+                                   BtcBalanceInfo actual) {
         assertEquals(expected.getAvailableBalance(), actual.getAvailableBalance());
         assertEquals(expected.getReservedBalance(), actual.getReservedBalance());
         assertEquals(expected.getTotalAvailableBalance(), actual.getTotalAvailableBalance());