I've fixed this now, by extending the case class to add the new functionality, leaving the original intact.