diff --git a/service-command/src/main/java/com/lanyuanxiaoyao/service/command/commands/ToolCommand.java b/service-command/src/main/java/com/lanyuanxiaoyao/service/command/commands/ToolCommand.java
index b613896..f25c6b0 100644
--- a/service-command/src/main/java/com/lanyuanxiaoyao/service/command/commands/ToolCommand.java
+++ b/service-command/src/main/java/com/lanyuanxiaoyao/service/command/commands/ToolCommand.java
@@ -40,9 +40,8 @@ public class ToolCommand {
                     if (hudiService.existsPath(hdfs)) {
                         String propertiesPath = StrUtil.format("{}/.hoodie/hoodie.properties", hdfs);
                         String content = hudiService.read(propertiesPath);
-                        if (content.contains("org.apache.hudi.common.model.OverwriteWithLatestAvroPayload") && content.contains("org.apache.hudi.keygen.SimpleKeyGenerator")) {
-                            content = content.replace("org.apache.hudi.common.model.OverwriteWithLatestAvroPayload", "org.apache.hudi.common.model.TraceOverwriteWithLatestAvroPayload");
-                            content = content.replace("org.apache.hudi.keygen.SimpleKeyGenerator", "org.apache.hudi.keygen.DefaultPartitionNameKeyGenerator");
+                        if (content.contains("org.apache.hudi.common.model.OverwriteWithLatestAvroPayload")) {
+                            content = content.replace("org.apache.hudi.common.model.OverwriteWithLatestAvroPayload", "org.apache.hudi.common.model.TraceEventTimeAvroPayload");
                             hudiService.write(propertiesPath, content, true);
                             logger.info("{} has rewrote", hdfs);
                             results.add(StrUtil.format("{} {}", meta.getJob().getId(), meta.getAlias()));
@@ -63,9 +62,8 @@ public class ToolCommand {
                     if (hudiService.existsPath(hdfs)) {
                         String propertiesPath = StrUtil.format("{}/.hoodie/hoodie.properties", hdfs);
                         String content = hudiService.read(propertiesPath);
-                        if (content.contains("org.apache.hudi.common.model.TraceOverwriteWithLatestAvroPayload") && content.contains("org.apache.hudi.keygen.DefaultPartitionNameKeyGenerator")) {
-                            content = content.replace("org.apache.hudi.common.model.TraceOverwriteWithLatestAvroPayload", "org.apache.hudi.common.model.OverwriteWithLatestAvroPayload");
-                            content = content.replace("org.apache.hudi.keygen.DefaultPartitionNameKeyGenerator", "org.apache.hudi.keygen.SimpleKeyGenerator");
+                        if (content.contains("org.apache.hudi.common.model.TraceEventTimeAvroPayload")) {
+                            content = content.replace("org.apache.hudi.common.model.TraceEventTimeAvroPayload", "org.apache.hudi.common.model.OverwriteWithLatestAvroPayload");
                             hudiService.write(propertiesPath, content, true);
                             logger.info("{} has rewrote", hdfs);
                             results.add(StrUtil.format("{} {}", meta.getJob().getId(), meta.getAlias()));