1 files changed, 5 insertions, 1 deletions
diff --git a/src/main/scala/xyz/driver/pdsuicommon/utils/CustomSwaggerJsonFormats.scala b/src/main/scala/xyz/driver/pdsuicommon/utils/CustomSwaggerJsonFormats.scala
index 004a914..55311ac 100644
--- a/src/main/scala/xyz/driver/pdsuicommon/utils/CustomSwaggerJsonFormats.scala
+++ b/src/main/scala/xyz/driver/pdsuicommon/utils/CustomSwaggerJsonFormats.scala
@@ -48,6 +48,8 @@ object CustomSwaggerJsonFormats {
     import xyz.driver.pdsuidomain.fakes.entities.export
     import xyz.driver.pdsuidomain.formats.json.sprayformats.export._
     import xyz.driver.pdsuidomain.formats.json.sprayformats.arm._
+    import xyz.driver.pdsuidomain.formats.json.sprayformats.slotarm._
+    import xyz.driver.pdsuidomain.formats.json.sprayformats.eligibilityarm._
     import xyz.driver.pdsuidomain.formats.json.sprayformats.criterion._
     import xyz.driver.pdsuidomain.formats.json.sprayformats.intervention._
     import xyz.driver.pdsuidomain.formats.json.sprayformats.hypothesis._
@@ -72,7 +74,9 @@ object CustomSwaggerJsonFormats {
       classOf[InterventionType]      -> interventionTypeFormat.write(nextInterventionType()),
       classOf[Hypothesis]            -> hypothesisFormat.write(nextHypothesis()),
       classOf[StudyDesign]           -> studyDesignFormat.write(nextStudyDesign()),
-      classOf[ExportTrialWithLabels] -> trialWithLabelsFormat.write(export.nextExportTrialWithLabels())
+      classOf[ExportTrialWithLabels] -> trialWithLabelsFormat.write(export.nextExportTrialWithLabels()),
+      classOf[EligibilityArmWithDiseases] -> eligibilityArmWithDiseasesWriter.write(nextEligibilityArmWithDiseases()),
+      classOf[SlotArm]                    -> slotArmFormat.write(nextSlotArm())
     )
   }