diff --git a/src/main/java/org/hbp/mip/controllers/ModelsApi.java b/src/main/java/org/hbp/mip/controllers/ModelsApi.java
index b8f6972c20125751e91e9c0e8b63d8074f0b2de4..e2f20f8fa5bcbb28920536fb2de9480e0c81fe27 100644
--- a/src/main/java/org/hbp/mip/controllers/ModelsApi.java
+++ b/src/main/java/org/hbp/mip/controllers/ModelsApi.java
@@ -91,6 +91,29 @@ public class ModelsApi {
             }
         }
 
+        for(Model model:models){
+            String ds_code = model.getDataset().getCode();
+
+            session = HibernateUtil.getSessionFactory().getCurrentSession();
+            Dataset dataset = null;
+            try{
+                session.beginTransaction();
+                dataset = (Dataset) session
+                        .createQuery("from Dataset where code= :code")
+                        .setString("code", ds_code)
+                        .uniqueResult();
+            } catch (Exception e)
+            {
+                if(session.getTransaction() != null)
+                {
+                    session.getTransaction().rollback();
+                    throw e;
+                }
+            }
+
+            model.setDataset(dataset);
+        }
+
         return new ResponseEntity<List<Model>>(HttpStatus.OK).ok(models);
     }