Add back hero-unit