diff --git a/flink-connector-oceanbase/pom.xml b/flink-connector-oceanbase/pom.xml
index 42b2286a..ee8d8124 100644
--- a/flink-connector-oceanbase/pom.xml
+++ b/flink-connector-oceanbase/pom.xml
@@ -44,7 +44,7 @@ under the License.
         <dependency>
             <groupId>com.oceanbase</groupId>
             <artifactId>ob-partition-calculator</artifactId>
-            <version>1.7.0-RELEASE</version>
+            <version>1.8.0-RELEASE</version>
         </dependency>
 
         <dependency>
diff --git a/flink-connector-oceanbase/src/main/java/com/oceanbase/connector/flink/connection/OceanBaseConnectionPool.java b/flink-connector-oceanbase/src/main/java/com/oceanbase/connector/flink/connection/OceanBaseConnectionPool.java
index b3abd321..a05846f2 100644
--- a/flink-connector-oceanbase/src/main/java/com/oceanbase/connector/flink/connection/OceanBaseConnectionPool.java
+++ b/flink-connector-oceanbase/src/main/java/com/oceanbase/connector/flink/connection/OceanBaseConnectionPool.java
@@ -21,7 +21,6 @@
 import com.oceanbase.connector.flink.dialect.OceanBaseOracleDialect;
 import com.oceanbase.partition.calculator.enums.ObServerMode;
 import com.oceanbase.partition.calculator.helper.TableEntryExtractor;
-import com.oceanbase.partition.calculator.helper.TableEntryExtractorV4;
 import com.oceanbase.partition.calculator.model.TableEntry;
 import com.oceanbase.partition.calculator.model.TableEntryKey;
 
@@ -116,17 +115,12 @@ public OceanBaseTablePartInfo getTablePartInfo() {
             }
             TableEntry tableEntry;
             try (Connection connection = getConnection()) {
-                if (getConnectionInfo().getVersion().isV4()) {
-                    tableEntry =
-                            new TableEntryExtractorV4()
-                                    .queryTableEntry(
-                                            connection, getConnectionInfo().getTableEntryKey());
-                } else {
-                    tableEntry =
-                            new TableEntryExtractor()
-                                    .queryTableEntry(
-                                            connection, getConnectionInfo().getTableEntryKey());
-                }
+                tableEntry =
+                        new TableEntryExtractor()
+                                .queryTableEntry(
+                                        connection,
+                                        getConnectionInfo().getTableEntryKey(),
+                                        getConnectionInfo().getVersion().isV4());
                 if (tableEntry == null) {
                     throw new RuntimeException(
                             "Failed to get table entry with key: "