diff --git a/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java b/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java
index fd75953..0a331d4 100644
--- a/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java
+++ b/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java
@@ -504,7 +504,6 @@ public class ModelTrainMngCoreService {
     return datasetRepository.findDatasetUid(datasetIds);
   }
 
-  @Transactional(readOnly = true)
   public List<Long> findModelDatasetMapp(Long modelId) {
     List<Long> datasetUids = new ArrayList<>();
     List<ModelDatasetMappEntity> entities = modelDatasetMapRepository.findByModelUid(modelId);
diff --git a/src/main/java/com/kamco/cd/training/postgres/repository/model/ModelDatasetMappRepositoryImpl.java b/src/main/java/com/kamco/cd/training/postgres/repository/model/ModelDatasetMappRepositoryImpl.java
index b43fdc7..1357a46 100644
--- a/src/main/java/com/kamco/cd/training/postgres/repository/model/ModelDatasetMappRepositoryImpl.java
+++ b/src/main/java/com/kamco/cd/training/postgres/repository/model/ModelDatasetMappRepositoryImpl.java
@@ -16,10 +16,10 @@ public class ModelDatasetMappRepositoryImpl implements ModelDatasetMappRepositor
 
   @Override
   public List<ModelDatasetMappEntity> findByModelUid(Long modelId) {
-    queryFactory
+    return queryFactory
         .select(modelDatasetMappEntity)
         .from(modelDatasetMappEntity)
-        .where(modelDatasetMappEntity.modelUid.eq(modelId));
-    return List.of();
+        .where(modelDatasetMappEntity.modelUid.eq(modelId))
+        .fetch();
   }
 }