diff --git a/dmp-backend/web/src/main/java/eu/eudat/models/data/dataset/DatasetOverviewModel.java b/dmp-backend/web/src/main/java/eu/eudat/models/data/dataset/DatasetOverviewModel.java index 1fe23c6a8..2ef2aeba1 100644 --- a/dmp-backend/web/src/main/java/eu/eudat/models/data/dataset/DatasetOverviewModel.java +++ b/dmp-backend/web/src/main/java/eu/eudat/models/data/dataset/DatasetOverviewModel.java @@ -3,6 +3,8 @@ package eu.eudat.models.data.dataset; import eu.eudat.data.entities.Dataset; import eu.eudat.models.DataModel; import eu.eudat.models.data.datasetprofile.DatasetProfileOverviewModel; +import eu.eudat.models.data.grant.GrantOverviewModel; +import eu.eudat.models.data.listingmodels.DataManagementPlanOverviewModel; import eu.eudat.models.data.listingmodels.UserInfoListingModel; import java.util.Date; @@ -17,8 +19,8 @@ public class DatasetOverviewModel implements DataModel users; - private String dmp; - private String grant; + private DataManagementPlanOverviewModel dmp; + private GrantOverviewModel grant; private String description; private Boolean isPublic; private Date modified; @@ -59,19 +61,19 @@ public class DatasetOverviewModel implements DataModel new UserInfoListingModel().fromDataModel(x)).collect(Collectors.toList()); - this.dmp = entity.getDmp().getLabel(); - this.grant = entity.getDmp().getGrant().getLabel(); + this.dmp = new DataManagementPlanOverviewModel().fromDataModel(entity.getDmp()); + this.grant = new GrantOverviewModel().fromDataModel(entity.getDmp().getGrant()); this.description = entity.getDescription(); this.isPublic = entity.getDmp().isPublic(); this.modified = entity.getModified();