diff --git a/SpringBootProject/src/main/java/fr/iut/sciencequest/sae/assemblers/DifficulteModelAssembler.java b/SpringBootProject/src/main/java/fr/iut/sciencequest/sae/assemblers/DifficulteModelAssembler.java
index 14703bb..1ccb851 100644
--- a/SpringBootProject/src/main/java/fr/iut/sciencequest/sae/assemblers/DifficulteModelAssembler.java
+++ b/SpringBootProject/src/main/java/fr/iut/sciencequest/sae/assemblers/DifficulteModelAssembler.java
@@ -1,6 +1,7 @@
 package fr.iut.sciencequest.sae.assemblers;
 
 import fr.iut.sciencequest.sae.dto.difficulte.DifficulteDTO;
+import fr.iut.sciencequest.sae.dto.difficulte.DifficulteSimpleDTO;
 import fr.iut.sciencequest.sae.entities.Difficulte;
 import jakarta.annotation.Nullable;
 import lombok.RequiredArgsConstructor;
@@ -10,16 +11,16 @@ import org.springframework.lang.NonNull;
 import org.springframework.stereotype.Component;
 
 @Component
-public class DifficulteModelAssembler extends RepresentationModelAssemblerSupport<Difficulte, DifficulteDTO> {
+public class DifficulteModelAssembler extends RepresentationModelAssemblerSupport<Difficulte, DifficulteSimpleDTO> {
     public DifficulteModelAssembler() {
-        super(Difficulte.class, DifficulteDTO.class);
+        super(Difficulte.class, DifficulteSimpleDTO.class);
     }
 
     @Override
     @NonNull
-    public DifficulteDTO toModel(@Nullable Difficulte entity) {
+    public DifficulteSimpleDTO toModel(@Nullable Difficulte entity) {
         ModelMapper mapper = new ModelMapper();
-        return mapper.map(entity, DifficulteDTO.class);
+        return mapper.map(entity, DifficulteSimpleDTO.class);
     }
 
 }