Replace deprecated get_mut with [] notation