diff --git a/lastleg-service/src/test/java/com/baiyue/lastleg/inboundplan/pda/gateway/HttpUnloadPhotoFileGatewayTest.java b/lastleg-service/src/test/java/com/baiyue/lastleg/inboundplan/pda/gateway/HttpUnloadPhotoFileGatewayTest.java
index 3149e00d5..734406198 100644
--- a/lastleg-service/src/test/java/com/baiyue/lastleg/inboundplan/pda/gateway/HttpUnloadPhotoFileGatewayTest.java
+++ b/lastleg-service/src/test/java/com/baiyue/lastleg/inboundplan/pda/gateway/HttpUnloadPhotoFileGatewayTest.java
@@ -23,10 +23,8 @@ class HttpUnloadPhotoFileGatewayTest {
         RestClient.Builder builder = RestClient.builder();
         server = MockRestServiceServer.bindTo(builder).build();
         gateway = new HttpUnloadPhotoFileGateway();
-        ReflectionTestUtils.setField(gateway, "restClientBuilder", builder);
-        ReflectionTestUtils.setField(gateway, "baseUrl", "http://file-service");
+        ReflectionTestUtils.setField(gateway, "client", builder.baseUrl("http://file-service").build());
         ReflectionTestUtils.setField(gateway, "internalToken", "internal-token");
-        gateway.initializeClient();
     }
 
     @Test
diff --git a/lastleg-service/src/test/java/com/baiyue/lastleg/inboundplan/pda/gateway/HttpWarehouseResourceGatewayTest.java b/lastleg-service/src/test/java/com/baiyue/lastleg/inboundplan/pda/gateway/HttpWarehouseResourceGatewayTest.java
index a4005d745..4b26ac3ca 100644
--- a/lastleg-service/src/test/java/com/baiyue/lastleg/inboundplan/pda/gateway/HttpWarehouseResourceGatewayTest.java
+++ b/lastleg-service/src/test/java/com/baiyue/lastleg/inboundplan/pda/gateway/HttpWarehouseResourceGatewayTest.java
@@ -20,10 +20,8 @@ class HttpWarehouseResourceGatewayTest {
         RestClient.Builder builder = RestClient.builder();
         server = MockRestServiceServer.bindTo(builder).build();
         gateway = new HttpWarehouseResourceGateway();
-        ReflectionTestUtils.setField(gateway, "restClientBuilder", builder);
-        ReflectionTestUtils.setField(gateway, "baseUrl", "http://warehouse");
+        ReflectionTestUtils.setField(gateway, "client", builder.baseUrl("http://warehouse").build());
         ReflectionTestUtils.setField(gateway, "internalToken", "test-token");
-        gateway.initializeClient();
     }
 
     @Test