Fix misnamed class