diff --git a/designer-base/src/main/java/com/fr/design/data/DesignTableDataManager.java b/designer-base/src/main/java/com/fr/design/data/DesignTableDataManager.java
index f724942524..35a539a50c 100644
--- a/designer-base/src/main/java/com/fr/design/data/DesignTableDataManager.java
+++ b/designer-base/src/main/java/com/fr/design/data/DesignTableDataManager.java
@@ -624,7 +624,7 @@ public abstract class DesignTableDataManager {
             XMLPrintWriter writer = XMLPrintWriter.create(out);
             // 把storeProcedure写成xml文件到out
             DataCoreXmlUtils.writeXMLStoreProcedure(writer, storeProcedure, null);
-            if (storeProcedure.getDataModelSize() > 0 && !storeProcedure.isFirstExpand()) {
+            if (storeProcedure.getDataModelList().size() > 0 && !storeProcedure.isFirstExpand()) {
                 return storeProcedure.getDataModelList().toArray(new ProcedureDataModel[0]);
             }
             ParameterProvider[] inParameters = DataOperator.getInstance().getStoreProcedureParameters(storeProcedure);
diff --git a/designer-base/src/main/java/com/fr/design/data/tabledata/wrapper/MultiResultTableDataNameWrapper.java b/designer-base/src/main/java/com/fr/design/data/tabledata/wrapper/MultiResultTableDataNameWrapper.java
index 6b42fa03b2..610aac3c58 100644
--- a/designer-base/src/main/java/com/fr/design/data/tabledata/wrapper/MultiResultTableDataNameWrapper.java
+++ b/designer-base/src/main/java/com/fr/design/data/tabledata/wrapper/MultiResultTableDataNameWrapper.java
@@ -69,7 +69,7 @@ public final class MultiResultTableDataNameWrapper implements TableDataWrapper {
     @Override
     public Icon getIcon() {
         for (TableDataNameObjectCreator creator : TableDataCreatorProducer.getInstance().createReportTableDataCreator()) {
-            if (creator.createObject().getClass() == this.tableData.getClass()) {
+            if (creator.createObject().getClass().isAssignableFrom(this.tableData.getClass())) {
                 return IOUtils.readIcon(creator.getIconPath());
             }
         }
diff --git a/designer-base/src/main/java/com/fr/design/data/tabledata/wrapper/MultiResultTableDataWrapper.java b/designer-base/src/main/java/com/fr/design/data/tabledata/wrapper/MultiResultTableDataWrapper.java
index 698b192208..6cce86fd20 100644
--- a/designer-base/src/main/java/com/fr/design/data/tabledata/wrapper/MultiResultTableDataWrapper.java
+++ b/designer-base/src/main/java/com/fr/design/data/tabledata/wrapper/MultiResultTableDataWrapper.java
@@ -164,7 +164,7 @@ public final class MultiResultTableDataWrapper implements TableDataWrapper {
     @Override
     public Icon getIcon() {
         for (TableDataNameObjectCreator creator : TableDataCreatorProducer.getInstance().createReportTableDataCreator()) {
-            if (creator.createObject().getClass() == this.tableData.getClass()) {
+            if (creator.createObject().getClass().isAssignableFrom(this.tableData.getClass())) {
                 return IOUtils.readIcon(creator.getIconPath());
             }
         }