1 files changed, 2 insertions, 6 deletions
diff --git a/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/rep/DocumentGen.scala b/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/rep/DocumentGen.scala
index 6d5330e..1ac75ab 100644
--- a/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/rep/DocumentGen.scala
+++ b/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/rep/DocumentGen.scala
@@ -75,12 +75,8 @@ object DocumentGen {
     )
   }
 
-  def nextDocumentType(): DocumentType = {
-    DocumentType(
-      id = nextLongId[DocumentType],
-      name = nextString()
-    )
-  }
+  def nextDocumentType(): DocumentType =
+    generators.oneOf(DocumentType.All: _*)
 
   def nextDocumentIssue(documentId: LongId[Document] = nextLongId): DocumentIssue = {
     val pages = nextStartAndEndPagesOption()