diff --git a/src/main/java/com/alibaba/excel/util/WriteHandlerUtils.java b/src/main/java/com/alibaba/excel/util/WriteHandlerUtils.java
index 57f0c49729f2a9632df694dee1caea7dd48332b2..044e7b2eb75dba2848377e352cb0b2159db23840 100644
--- a/src/main/java/com/alibaba/excel/util/WriteHandlerUtils.java
+++ b/src/main/java/com/alibaba/excel/util/WriteHandlerUtils.java
@@ -234,6 +234,6 @@ public class WriteHandlerUtils {
         } else {
             writeHandlerMap = writeContext.currentWriteHolder().writeHandlerMap();
         }
-        return writeHandlerMap.get(WorkbookWriteHandler.class);
+        return writeHandlerMap.get(clazz);
     }
 }