1 files changed, 2 insertions, 2 deletions
diff --git a/src/main/scala/xyz/driver/pdsuidomain/entities/EligibilityArm.scala b/src/main/scala/xyz/driver/pdsuidomain/entities/EligibilityArm.scala
index 329c15c..50b49ad 100644
--- a/src/main/scala/xyz/driver/pdsuidomain/entities/EligibilityArm.scala
+++ b/src/main/scala/xyz/driver/pdsuidomain/entities/EligibilityArm.scala
@@ -2,9 +2,9 @@ package xyz.driver.pdsuidomain.entities
 
 import java.time.LocalDateTime
 
+import xyz.driver.entities.patient.CancerType
 import xyz.driver.pdsuicommon.domain.{LongId, StringId}
 import xyz.driver.pdsuicommon.logging._
-import xyz.driver.pdsuidomain.entities.Trial.Condition
 
 final case class EligibilityArm(id: LongId[EligibilityArm],
                                 name: String,
@@ -20,7 +20,7 @@ object EligibilityArm {
   }
 }
 
-final case class EligibilityArmDisease(eligibilityArmId: LongId[EligibilityArm], disease: Condition)
+final case class EligibilityArmDisease(eligibilityArmId: LongId[EligibilityArm], disease: CancerType)
 
 object EligibilityArmDisease {