Added condition for hero unit