When I rewrite that code using

sealed trait

and `case class`/`case object`, it seems to work