added a missing case class