2 files changed, 6 insertions, 5 deletions
diff --git a/src/main/scala/xyz/driver/pdsuidomain/entities/Document.scala b/src/main/scala/xyz/driver/pdsuidomain/entities/Document.scala
index 1f73184..839fead 100644
--- a/src/main/scala/xyz/driver/pdsuidomain/entities/Document.scala
+++ b/src/main/scala/xyz/driver/pdsuidomain/entities/Document.scala
@@ -14,6 +14,8 @@ import xyz.driver.pdsuicommon.validation.Validators
 import xyz.driver.pdsuicommon.validation.Validators.Validator
 import xyz.driver.pdsuidomain.entities.Document.Meta
 
+import scalaz.Equal
+
 final case class ProviderType(id: LongId[ProviderType], name: String)
 
 object ProviderType {
@@ -124,6 +126,8 @@ object DocumentType {
     }
   }
 
+  implicit def equal: Equal[DocumentType] = Equal.equal[DocumentType](_ == _)
+
   implicit def toPhiString(x: DocumentType): PhiString = {
     import x._
     phi"DocumentType(id=$id, name=${Unsafe(name)})"
diff --git a/src/main/scala/xyz/driver/pdsuidomain/entities/export/trial/ExportTrialWithLabels.scala b/src/main/scala/xyz/driver/pdsuidomain/entities/export/trial/ExportTrialWithLabels.scala
index 60b74ff..cf55694 100644
--- a/src/main/scala/xyz/driver/pdsuidomain/entities/export/trial/ExportTrialWithLabels.scala
+++ b/src/main/scala/xyz/driver/pdsuidomain/entities/export/trial/ExportTrialWithLabels.scala
@@ -27,7 +27,7 @@ object ExportTrialWithLabels {
   def fromRaw(rawData: List[RawTrialLabel]): ExportTrialWithLabels = {
     val trials: Set[StringId[Trial]] = rawData.map(_.nctId)(breakOut)
 
-    assert(trials.size == 1, "There are more than one trials in the rawData")
+    assert(trials.size == 1, "There are more than one trial in the rawData")
     val trial = rawData.head
 
     ExportTrialWithLabels(
@@ -43,9 +43,7 @@ object ExportTrialWithLabels {
             ExportTrialArm(armId, rawTrials.head.armName)
         }(breakOut),
       criteria = rawData
-        .groupBy { x =>
-          (x.criterionId, x.labelId)
-        }
+        .groupBy(x => (x.criterionId, x.labelId))
         .map {
           case (_, rawTrialLabels) =>
             val armIds = rawTrialLabels.map(_.criterionArmId).toSet
@@ -53,5 +51,4 @@ object ExportTrialWithLabels {
         }(breakOut)
     )
   }
-
 }