fixed unit display bugs