1 files changed, 6 insertions, 14 deletions
diff --git a/src/main/scala/xyz/driver/pdsuidomain/entities/DocumentHistory.scala b/src/main/scala/xyz/driver/pdsuidomain/entities/DocumentHistory.scala
index 5c96f94..089658e 100644
--- a/src/main/scala/xyz/driver/pdsuidomain/entities/DocumentHistory.scala
+++ b/src/main/scala/xyz/driver/pdsuidomain/entities/DocumentHistory.scala
@@ -8,8 +8,6 @@ import xyz.driver.pdsuicommon.logging._
 import xyz.driver.pdsuicommon.utils.Utils
 import xyz.driver.pdsuidomain.entities.DocumentHistory.{Action, _}
 
-import scala.collection.immutable
-
 object DocumentHistory {
 
   implicit def toPhiString(x: DocumentHistory): PhiString = {
@@ -25,18 +23,15 @@ object DocumentHistory {
     case object Flag    extends State
     case object New     extends State
 
-    private implicit def stateToName(state: State): (State, String) = {
-      state -> state.toString
-    }
-
-    private val stateToName = immutable.Map[State, String](
+    val All: Set[State] = Set(
       State.Extract,
       State.Review,
       State.Flag,
       State.New
     )
 
-    val All: Set[State] = stateToName.keySet
+    private val stateToName: Map[State, String] =
+      All.map(s => s -> s.toString).toMap
 
     val fromString: PartialFunction[String, State] =
       for ((k, v) <- stateToName) yield (v, k)
@@ -66,11 +61,7 @@ object DocumentHistory {
     case object CreatedDocument extends Action
     case object ReadDocument    extends Action
 
-    private implicit def stateToName(action: Action): (Action, String) = {
-      action -> action.toString
-    }
-
-    private val actionToName = immutable.Map[Action, String](
+    val All: Set[Action] = Set(
       Action.Start,
       Action.Submit,
       Action.Unassign,
@@ -82,7 +73,8 @@ object DocumentHistory {
       Action.ReadDocument
     )
 
-    val All: Set[Action] = actionToName.keySet
+    private val actionToName: Map[Action, String] =
+      All.map(a => a -> a.toString).toMap
 
     val fromString: PartialFunction[String, Action] =
       for ((k, v) <- actionToName) yield (v, k)