diff --git a/src/main/java/com/kamco/cd/training/train/dto/ModelTrainLinkDto.java b/src/main/java/com/kamco/cd/training/train/dto/ModelTrainLinkDto.java
index d866200..19043f1 100644
--- a/src/main/java/com/kamco/cd/training/train/dto/ModelTrainLinkDto.java
+++ b/src/main/java/com/kamco/cd/training/train/dto/ModelTrainLinkDto.java
@@ -1,10 +1,14 @@
 package com.kamco.cd.training.train.dto;
 
+import lombok.AllArgsConstructor;
 import lombok.Getter;
+import lombok.NoArgsConstructor;
 import lombok.Setter;
 
 @Getter
 @Setter
+@NoArgsConstructor
+@AllArgsConstructor
 public class ModelTrainLinkDto {
 
   private Long modelId;