diff --git a/src/main/java/com/fanruan/api/design/DesignKit.java b/src/main/java/com/fanruan/api/design/DesignKit.java
index 2ff10ec..4b95d64 100644
--- a/src/main/java/com/fanruan/api/design/DesignKit.java
+++ b/src/main/java/com/fanruan/api/design/DesignKit.java
@@ -1,6 +1,8 @@
 package com.fanruan.api.design;
 
+import com.fr.base.io.BaseBook;
 import com.fr.design.DesignerEnvManager;
+import com.fr.design.mainframe.JTemplateProvider;
 import com.fr.design.mainframe.WidgetPropertyPane;
 import com.fr.base.TableData;
 import com.fr.design.DesignModelAdapter;
@@ -76,12 +78,12 @@ public class DesignKit {
 
     /**
      * 获取指定模板的ID
-     * 
+     *
      * @param template 指定模板
      * @return 模板ID
      */
-    public static String getTemplateId(JTemplate<?, ?> template) {
-        return template.getTarget().getTemplateID();
+    public static String getTemplateId(JTemplateProvider template) {
+        return ((BaseBook) template.getTarget()).getTemplateID();
     }
 
     /**