1 files changed, 2 insertions, 2 deletions
diff --git a/src/main/scala/xyz/driver/pdsuidomain/services/fake/FakeTrialService.scala b/src/main/scala/xyz/driver/pdsuidomain/services/fake/FakeTrialService.scala
index cb7f8bc..37c3de5 100644
--- a/src/main/scala/xyz/driver/pdsuidomain/services/fake/FakeTrialService.scala
+++ b/src/main/scala/xyz/driver/pdsuidomain/services/fake/FakeTrialService.scala
@@ -69,13 +69,13 @@ class FakeTrialService extends TrialService {
     ExportTrialWithLabels(
       StringId[Trial]("NCT" + generators.nextInt(999999).toString),
       UuidId[Trial](generators.nextUuid()),
-      generators.oneOf("adenocarcinoma", "breast", "prostate"),
       LocalDateTime.now(),
       labelVersion = 1L,
       generators.listOf(
         new ExportTrialArm(
           LongId[Arm](generators.nextInt(999999).toLong),
-          generators.nextName().value
+          generators.nextName().value,
+          generators.listOf(generators.oneOf("adenocarcinoma", "breast", "prostate"))
         )),
       generators.listOf(
         new ExportTrialLabelCriterion(