diff --git a/types/wlr_input_method_v2.c b/types/wlr_input_method_v2.c
index 4b42916de8..7082fb2697 100644
--- a/types/wlr_input_method_v2.c
+++ b/types/wlr_input_method_v2.c
@@ -230,6 +230,7 @@ void wlr_input_method_keyboard_grab_v2_set_keyboard(
 	}
 	im_keyboard_grab_send_repeat_info(im_keyboard_grab, keyboard);
 
+	im_keyboard_grab->keyboard = keyboard;
 	im_keyboard_grab->keymap_listener.notify = handle_keyboard_keymap;
 	wl_signal_add(&keyboard->events.keymap,
 		&im_keyboard_grab->keymap_listener);