fix test case class