fix unit display error