private void AddEntityToRecheckVisualAfterLoadList(SmartEntity entity, Buff buff) { BuildingComponent buildingComp = entity.BuildingComp; TroopComponent troopComp = entity.TroopComp; BuffComponent buffComp = entity.BuffComp; if (buildingComp != null) { buffComp.RegisterForVisualReAddOnBuilding(buff); } else if (troopComp != null) { buffComp.RegisterForVisualReAddOnTroop(buff); } }