Fix display unit selection