diff --git a/lib/generate_functions.py b/lib/generate_functions.py
index 2276707..091ec12 100644
--- a/lib/generate_functions.py
+++ b/lib/generate_functions.py
@@ -52,7 +52,10 @@ def fix_enums(arg_name, arg_type, func_name):
         if arg_name == "key":
             arg_type = "KeyboardKey"
         elif arg_name == "button":
-            arg_type = "MouseButton"
+            if "Gamepad" in func_name:
+                arg_type = "GamepadButton"
+            else:
+                arg_type = "MouseButton"
         elif arg_name == "mode" and func_name == "SetCameraMode":
             arg_type = "CameraMode"
         elif arg_name == "gesture":
diff --git a/lib/raylib-wa.zig b/lib/raylib-wa.zig
index 2f79c3e..9b331df 100644
--- a/lib/raylib-wa.zig
+++ b/lib/raylib-wa.zig
@@ -9,6 +9,7 @@ const Camera3D = rl.Camera3D;
 const CameraMode = rl.CameraMode;
 const Color = rl.Color;
 const Font = rl.Font;
+const GamepadButton = rl.GamepadButton;
 const Gestures = rl.Gestures;
 const GlyphInfo = rl.GlyphInfo;
 const Image = rl.Image;
@@ -22,7 +23,6 @@ const Mesh = rl.Mesh;
 const Model = rl.Model;
 const ModelAnimation = rl.ModelAnimation;
 const MouseButton = rl.MouseButton;
-const GamepadButton = rl.GamepadButton;
 const Music = rl.Music;
 const NPatchInfo = rl.NPatchInfo;
 const Quaternion = rl.Quaternion;