diff --git a/hudi-client/hudi-client-common/src/main/java/org/apache/hudi/table/upgrade/FourToFiveUpgradeHandler.java b/hudi-client/hudi-client-common/src/main/java/org/apache/hudi/table/upgrade/FourToFiveUpgradeHandler.java
index 4d7c5b8b6df6a..2adddf36df503 100644
--- a/hudi-client/hudi-client-common/src/main/java/org/apache/hudi/table/upgrade/FourToFiveUpgradeHandler.java
+++ b/hudi-client/hudi-client-common/src/main/java/org/apache/hudi/table/upgrade/FourToFiveUpgradeHandler.java
@@ -77,7 +77,7 @@ public Map<ConfigProperty, String> upgrade(HoodieWriteConfig config, HoodieEngin
 
   private boolean hasDefaultPartitionPath(HoodieWriteConfig config, HoodieTable  table) throws IOException {
     HoodieTableConfig tableConfig = table.getMetaClient().getTableConfig();
-    if (!tableConfig.getPartitionFields().isPresent()) {
+    if (!tableConfig.isTablePartitioned()) {
       return false;
     }
     String checkPartitionPath = DEPRECATED_DEFAULT_PARTITION_PATH;