diff --git a/src/includes.hpp b/src/includes.hpp
index 8b00b394..6793604b 100644
--- a/src/includes.hpp
+++ b/src/includes.hpp
@@ -87,6 +87,7 @@ extern "C" {
 #include "../wlroots/include/wlr/render/wlr_texture.h"
 #include "../wlroots/include/wlr/types/wlr_pointer_constraints_v1.h"
 #include "../wlroots/include/wlr/types/wlr_relative_pointer_v1.h"
+#include "../wlroots/include/wlr/interfaces/wlr_keyboard.h"
 }
 
 #undef class
diff --git a/src/managers/input/InputManager.cpp b/src/managers/input/InputManager.cpp
index 61160be0..003535f7 100644
--- a/src/managers/input/InputManager.cpp
+++ b/src/managers/input/InputManager.cpp
@@ -309,7 +309,7 @@ void CInputManager::setKeyboardLayout() {
     }
 
     if (wlrMods.locked != 0) {
-        wlr_seat_keyboard_notify_modifiers(g_pCompositor->m_sSeat.seat, &wlrMods);
+        wlr_keyboard_notify_modifiers(g_pInputManager->m_pActiveKeyboard->keyboard->keyboard, 0, 0, wlrMods.locked, 0);
     }
 
     xkb_keymap_unref(KEYMAP);