diff --git a/src/main/java/nl/tudelft/ewi/walkytalky/dto/ScheduleDto.java b/src/main/java/nl/tudelft/ewi/walkytalky/dto/ScheduleDto.java
index 612cd2144c9cdf8f66e062338ff3075843cfa325..0cacee820eea19bdc097bec60786832c214c0349 100644
--- a/src/main/java/nl/tudelft/ewi/walkytalky/dto/ScheduleDto.java
+++ b/src/main/java/nl/tudelft/ewi/walkytalky/dto/ScheduleDto.java
@@ -23,15 +23,13 @@ import java.time.LocalDateTime;
 
 import javax.validation.constraints.NotNull;
 
-import lombok.AllArgsConstructor;
-import lombok.Builder;
-import lombok.Data;
-import lombok.NoArgsConstructor;
+import lombok.*;
 import nl.tudelft.ewi.walkytalky.model.Schedule;
 import nl.tudelft.ewi.walkytalky.validator.ValidScheduleDto;
 
 import org.springframework.format.annotation.DateTimeFormat;
 
+@EqualsAndHashCode(callSuper = true)
 @ValidScheduleDto
 @Data
 @Builder
diff --git a/src/main/java/nl/tudelft/ewi/walkytalky/dto/UnitPatchDto.java b/src/main/java/nl/tudelft/ewi/walkytalky/dto/UnitPatchDto.java
index 0066d69fe5cd5c60674673032ce77f11a70ffe47..c86048b54cd2ce9ea6a799e7b24a9a993709825a 100644
--- a/src/main/java/nl/tudelft/ewi/walkytalky/dto/UnitPatchDto.java
+++ b/src/main/java/nl/tudelft/ewi/walkytalky/dto/UnitPatchDto.java
@@ -21,13 +21,11 @@ package nl.tudelft.ewi.walkytalky.dto;
 
 import java.util.List;
 
-import lombok.AllArgsConstructor;
-import lombok.Builder;
-import lombok.Data;
-import lombok.NoArgsConstructor;
+import lombok.*;
 import nl.tudelft.ewi.walkytalky.model.Patient;
 import nl.tudelft.ewi.walkytalky.model.Unit;
 
+@EqualsAndHashCode(callSuper = true)
 @Data
 @Builder
 @AllArgsConstructor