diff --git a/addons/imgui-godot/ImGuiGodot/ImGuiLayer.cs b/addons/imgui-godot/ImGuiGodot/ImGuiLayer.cs
index a5c8348..21f9ea6 100644
--- a/addons/imgui-godot/ImGuiGodot/ImGuiLayer.cs
+++ b/addons/imgui-godot/ImGuiGodot/ImGuiLayer.cs
@@ -209,7 +209,7 @@ public partial class ImGuiLayer : CanvasLayer
     {
         if (ImGuiGD.ProcessInput(e))
         {
-            GetViewport().SetInputAsHandled();
+            _window.SetInputAsHandled();
         }
     }