diff --git a/src/main/java/com/blessedbits/SchoolHub/controllers/ClassesController.java b/src/main/java/com/blessedbits/SchoolHub/controllers/ClassesController.java index 65d6beb..2668c98 100644 --- a/src/main/java/com/blessedbits/SchoolHub/controllers/ClassesController.java +++ b/src/main/java/com/blessedbits/SchoolHub/controllers/ClassesController.java @@ -80,7 +80,8 @@ public ResponseEntity createClass( classEntity.setHomeroomTeacher(teacher); classEntity.setSchool(school); try { - return new ResponseEntity<>(classRepository.save(classEntity), HttpStatus.CREATED); + classRepository.save(classEntity); + return new ResponseEntity<>(ClassMapper.INSTANCE.toClassDto(classEntity, List.of("homeroomTeacher")), HttpStatus.CREATED); } catch (Exception e) { System.out.println(e.getMessage()); return new ResponseEntity<>(