diff --git a/wherehows-common/src/main/java/wherehows/common/schemas/DatasetComplianceRecord.java b/wherehows-common/src/main/java/wherehows/common/schemas/DatasetComplianceRecord.java
index d377685259..f06220169a 100644
--- a/wherehows-common/src/main/java/wherehows/common/schemas/DatasetComplianceRecord.java
+++ b/wherehows-common/src/main/java/wherehows/common/schemas/DatasetComplianceRecord.java
@@ -14,7 +14,6 @@
 package wherehows.common.schemas;
 
 import java.util.List;
-import java.util.Map;
 
 
 public class DatasetComplianceRecord extends AbstractRecord {
@@ -22,7 +21,7 @@ public class DatasetComplianceRecord extends AbstractRecord {
   Integer datasetId;
   String datasetUrn;
   String complianceType;
-  List<DatasetEntityRecord> compliancePurgeEntities;
+  List<DatasetPurgeEntityRecord> compliancePurgeEntities;
   Long modifiedTime;
 
   @Override
@@ -63,11 +62,11 @@ public class DatasetComplianceRecord extends AbstractRecord {
     this.complianceType = complianceType;
   }
 
-  public List<DatasetEntityRecord> getCompliancePurgeEntities() {
+  public List<DatasetPurgeEntityRecord> getCompliancePurgeEntities() {
     return compliancePurgeEntities;
   }
 
-  public void setCompliancePurgeEntities(List<DatasetEntityRecord> compliancePurgeEntities) {
+  public void setCompliancePurgeEntities(List<DatasetPurgeEntityRecord> compliancePurgeEntities) {
     this.compliancePurgeEntities = compliancePurgeEntities;
   }
 
diff --git a/wherehows-common/src/main/java/wherehows/common/schemas/DatasetEntityRecord.java b/wherehows-common/src/main/java/wherehows/common/schemas/DatasetPurgeEntityRecord.java
similarity index 70%
rename from wherehows-common/src/main/java/wherehows/common/schemas/DatasetEntityRecord.java
rename to wherehows-common/src/main/java/wherehows/common/schemas/DatasetPurgeEntityRecord.java
index 0b208c1fed..988811cdb7 100644
--- a/wherehows-common/src/main/java/wherehows/common/schemas/DatasetEntityRecord.java
+++ b/wherehows-common/src/main/java/wherehows/common/schemas/DatasetPurgeEntityRecord.java
@@ -16,17 +16,19 @@ package wherehows.common.schemas;
 import java.util.List;
 
 
-public class DatasetEntityRecord extends AbstractRecord {
+public class DatasetPurgeEntityRecord extends AbstractRecord {
 
   String identifierType;
   String identifierField;
+  String logicalType;
+  Boolean isSubject;
 
   @Override
   public List<Object> fillAllFields() {
     return null;
   }
 
-  public DatasetEntityRecord() {
+  public DatasetPurgeEntityRecord() {
   }
 
   public String getIdentifierType() {
@@ -44,4 +46,20 @@ public class DatasetEntityRecord extends AbstractRecord {
   public void setIdentifierField(String identifierField) {
     this.identifierField = identifierField;
   }
+
+  public String getLogicalType() {
+    return logicalType;
+  }
+
+  public void setLogicalType(String logicalType) {
+    this.logicalType = logicalType;
+  }
+
+  public Boolean getIsSubject() {
+    return isSubject;
+  }
+
+  public void setIsSubject(Boolean isSubject) {
+    this.isSubject = isSubject;
+  }
 }