diff --git a/Hazel/src/Hazel/Core/Input.h b/Hazel/src/Hazel/Core/Input.h
index bd6fbe2..4b72088 100644
--- a/Hazel/src/Hazel/Core/Input.h
+++ b/Hazel/src/Hazel/Core/Input.h
@@ -7,6 +7,7 @@
 #include <utility>
 
 #include "Core.h"
+#include "KeyCodes.h"
 
 namespace Hazel
 {
@@ -14,9 +15,9 @@ namespace Hazel
     public:
         virtual ~Input() = default;
 
-        static bool IsKeyPressed(int key);
+        static bool IsKeyPressed(KeyCode key);
 
-        static bool IsMouseButtonPressed(int button);
+        static bool IsMouseButtonPressed(MouseButton button);
         static std::pair<float, float> GetMousePosition();
 
         static float GetMouseX();