Minor fix for root class