1 files changed, 1 insertions, 8 deletions
diff --git a/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/trialcuration.scala b/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/trialcuration.scala
index 5bcf84f..84dcb88 100644
--- a/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/trialcuration.scala
+++ b/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/trialcuration.scala
@@ -2,7 +2,6 @@ package xyz.driver.pdsuidomain.fakes.entities
 
 import xyz.driver.core.generators._
 import xyz.driver.entities.labels.{Label, LabelCategory}
-import xyz.driver.entities.patient.CancerType
 import xyz.driver.pdsuicommon.domain.{LongId, User}
 import xyz.driver.pdsuidomain.entities._
 import xyz.driver.pdsuidomain.services.CriterionService.RichCriterion
@@ -156,15 +155,9 @@ object trialcuration {
     trialId = nextStringId
   )
 
-  def nextPatientCancerType(): CancerType = generators.oneOf[CancerType](
-    CancerType.Lung,
-    CancerType.Breast,
-    CancerType.Prostate
-  )
-
   def nextEligibilityArmDisease(): EligibilityArmDisease = EligibilityArmDisease(
     eligibilityArmId = nextLongId,
-    disease = nextPatientCancerType()
+    disease = nextCancerType
   )
 
   private def nextEligibilityArmDiseaseCollection(count: Int): Seq[EligibilityArmDisease] =