fix buttons on hero unit