diff --git a/Engine.Systems/Input/IButtonInputs.cs b/Engine.Systems/Input/IButtonInputs.cs
index 3501ee3..75cd678 100644
--- a/Engine.Systems/Input/IButtonInputs.cs
+++ b/Engine.Systems/Input/IButtonInputs.cs
@@ -7,10 +7,10 @@ public interface IButtonInputs<T> : IHasStateEnable
     Event<IButtonInputs<T>, ButtonCallbackArguments> OnAnyButtonPressed { get; }
     Event<IButtonInputs<T>, ButtonCallbackArguments> OnAnyButtonReleased { get; }
 
-    void RegisterOnPress(T button, ButtonCallbackArguments callback);
-    void UnregisterOnPress(T button, ButtonCallbackArguments callback);
-    void RegisterOnRelease(T button, ButtonCallbackArguments callback);
-    void UnregisterOnRelease(T button, ButtonCallbackArguments callback);
+    void RegisterOnPress(T button, Event<IButtonInputs<T>, ButtonCallbackArguments>.EventHandler callback);
+    void UnregisterOnPress(T button, Event<IButtonInputs<T>, ButtonCallbackArguments>.EventHandler callback);
+    void RegisterOnRelease(T button, Event<IButtonInputs<T>, ButtonCallbackArguments>.EventHandler callback);
+    void UnregisterOnRelease(T button, Event<IButtonInputs<T>, ButtonCallbackArguments>.EventHandler callback);
 
     bool IsPressed(T button);