Minor fix for dh class