diff --git a/src/input/ime.c b/src/input/ime.c
index cf8cd2d9..f6e0ac99 100644
--- a/src/input/ime.c
+++ b/src/input/ime.c
@@ -304,8 +304,8 @@ handle_keyboard_grab_destroy(struct wl_listener *listener, void *data)
 {
 	struct input_method_relay *relay =
 		wl_container_of(listener, relay, keyboard_grab_destroy);
-	struct wlr_input_method_keyboard_grab_v2 *keyboard_grab =
-		relay->input_method->keyboard_grab;
+	struct wlr_input_method_keyboard_grab_v2 *keyboard_grab = data;
+	assert(keyboard_grab);
 
 	wl_list_remove(&relay->keyboard_grab_destroy.link);