Renamed class in unit test