Fix test case class (add toList in toSource function)