diff --git a/hudi-utilities/src/main/java/org/apache/hudi/utilities/keygen/TimestampBasedKeyGenerator.java b/hudi-utilities/src/main/java/org/apache/hudi/utilities/keygen/TimestampBasedKeyGenerator.java
index c24ddc8494554..b8019b476a34b 100644
--- a/hudi-utilities/src/main/java/org/apache/hudi/utilities/keygen/TimestampBasedKeyGenerator.java
+++ b/hudi-utilities/src/main/java/org/apache/hudi/utilities/keygen/TimestampBasedKeyGenerator.java
@@ -103,7 +103,7 @@ public HoodieKey getKey(GenericRecord record) {
         unixTime = ((Float) partitionVal).longValue();
       } else if (partitionVal instanceof Long) {
         unixTime = (Long) partitionVal;
-      } else if (partitionVal instanceof String) {
+      } else if (partitionVal instanceof CharSequence) {
         unixTime = inputDateFormat.parse(partitionVal.toString()).getTime() / 1000;
       } else {
         throw new HoodieNotSupportedException(