fixed unit display (issue #1)