@@ -2705,6 +2705,9 @@ impl ui::menu::Item for MappableCommand {
2707pub fn command_palette(cx: &mut Context) {
2707pub fn command_palette(cx: &mut Context) {
2708+ let register = cx.register;
2709+ let count = cx.count;
2708 cx.callback = Some(Box::new(
2711 cx.callback = Some(Box::new(
2709 move |compositor: &mut Compositor, cx: &mut compositor::Context| {
2712 move |compositor: &mut Compositor, cx: &mut compositor::Context| {
2710 let keymap = compositor.find::<ui::EditorView>().unwrap().keymaps.map()
2713 let keymap = compositor.find::<ui::EditorView>().unwrap().keymaps.map()
@@ -2722,8 +2725,8 @@ pub fn command_palette(cx: &mut Context) {
2723 let picker = Picker::new(commands, keymap, move |cx, command, _action| {
2726 let picker = Picker::new(commands, keymap, move |cx, command, _action| {
2724 let mut ctx = Context {
2727 let mut ctx = Context {
2725− register: None,
2728+ register,
2726− count: std::num::NonZeroUsize::new(1),
2729+ count,
2727 editor: cx.editor,
2730 editor: cx.editor,
2728 callback: None,
2731 callback: None,
2729 on_next_key_callback: None,
2732 on_next_key_callback: None,