Referenced mutable resource with AppState

[EvoTeamDevelop]

Hi! I am new to the Rust and Bevy and I have stuck on case.

I am trying to create a separate plugin for my application, which will be responsible for menu creation using egui.
I have created in such way that I pass Vector with Buttons structure, which have handler (I am passing it as mutable closure).
But the problem is that loose state_machine ownership, when I am trying to add 2nd handler.
As I got it, I need to use reference, but I am not sure if it is possible in Bevy to get reference for mutable resource with state machine.
Can somebody advice how to solve it? Thank you!

bevy system:
fn ui_menu(
    egui_context: ResMut<EguiContext>,
    mut state_machine: ResMut<State<AppStates>>,
) {
    // Gets a mutable reference to the resource of the given type, if it exists. Otherwise returns None Resources are “unique” data of a given type.
    let mut buttons = vec![];

    // let refState =
    buttons.push(Button{
        title: "New training".to_string(),
        handler: Box::new(|| {
            state_machine.set(AppStates::NewTraining).unwrap(); // here closure captures state
        })
    });

    buttons.push(Button{
        title: "Add word".to_string(),
        handler: Box::new(|| {
            state_machine.set(AppStates::AddWord).unwrap(); // here I cannot already use it
        })
    });