diff --git a/integration_tests/client-library/java/src/test/java/com/risingwave/TestSubscription.java b/integration_tests/client-library/java/src/test/java/com/risingwave/TestSubscription.java
index 8dffc020e52cf..0c834e23806b2 100644
--- a/integration_tests/client-library/java/src/test/java/com/risingwave/TestSubscription.java
+++ b/integration_tests/client-library/java/src/test/java/com/risingwave/TestSubscription.java
@@ -1,6 +1,5 @@
 package com.risingwave;
 
-import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.Test;
 
 import java.sql.Connection;
diff --git a/integration_tests/client-library/java/src/test/java/com/risingwave/TestUtils.java b/integration_tests/client-library/java/src/test/java/com/risingwave/TestUtils.java
index cbb1eb0e65c28..245c2f36a06d2 100644
--- a/integration_tests/client-library/java/src/test/java/com/risingwave/TestUtils.java
+++ b/integration_tests/client-library/java/src/test/java/com/risingwave/TestUtils.java
@@ -7,7 +7,8 @@
 
 public class TestUtils {
     public static Connection establishConnection() throws SQLException {
-        final String url = "jdbc:postgresql://localhost:4566/dev?preferQueryMode=simple";
+        // TODO: remove preferQueryMode=simple.
+        final String url = "jdbc:postgresql://risingwave-standalone:4566/dev?preferQueryMode=simple";
         final String user = "root";
         final String password = "";