diff --git a/src/modules/ecore_imf/wayland/wayland_imcontext.c b/src/modules/ecore_imf/wayland/wayland_imcontext.c
index 2464278779..1c92cca5e6 100644
--- a/src/modules/ecore_imf/wayland/wayland_imcontext.c
+++ b/src/modules/ecore_imf/wayland/wayland_imcontext.c
@@ -776,7 +776,10 @@ wayland_im_context_client_window_set(Ecore_IMF_Context *ctx,
    EINA_LOG_DOM_INFO(_ecore_imf_wayland_log_dom, "client window set (window: %p)", window);
 
    if (window != NULL)
-     imcontext->window = ecore_wl2_window_find(ewd, (Ecore_Window)window);
+     {
+        imcontext->window =
+          ecore_wl2_display_window_find(ewd, (Ecore_Window)window);
+     }
 }
 
 EAPI void