diff --git a/dmp-backend/web/src/main/java/eu/eudat/logic/managers/DataManagementPlanManager.java b/dmp-backend/web/src/main/java/eu/eudat/logic/managers/DataManagementPlanManager.java
index 254d941e9..5ae417d25 100644
--- a/dmp-backend/web/src/main/java/eu/eudat/logic/managers/DataManagementPlanManager.java
+++ b/dmp-backend/web/src/main/java/eu/eudat/logic/managers/DataManagementPlanManager.java
@@ -473,8 +473,7 @@ public class DataManagementPlanManager {
             if (dmp1.getModified().getTime() != dataManagementPlan.getModified().getTime()) {
                 throw new Exception("Another user have already edit that DMP.");
             }
-            List<Dataset> datasetList = new ArrayList<>(dmp1.getDataset());
-            for (Dataset dataset : datasetList) {
+            for (DatasetWizardModel dataset : dataManagementPlan.getDatasets()) {
                 if (dataManagementPlan.getProfiles().stream().filter(associatedProfile -> dataset.getProfile().getId().equals(associatedProfile.getId())).findAny().orElse(null) == null)
                     throw new Exception("Dataset Template for Dataset Description is missing from the DMP.");
             }