Minor fix for broken unit frames