fix bug display unit