Как при клике на кнопку вызвать функцию нужного объекта в Unity?

В игре есть юниты, которые создаются из префаба. У каждого юнита есть оружие. Всего три типа оружия. Если дружественный юнит бьет врага с таким же типом оружия, то открывается экран с тремя кнопками для выбора другого типа оружия (принцип, как в Камень, ножницы, бумага). Проблема в том, чтобы при нажатии кнопки заменить оружие именно для выбранного юнита. Как мне передать объект, для которого это должно срабатывать?