1 files changed, 6 insertions, 1 deletions

diff --git a/firmware/drivers/button.h b/firmware/drivers/button.h
index a8cc71b875..ef8984466e 100644
--- a/firmware/drivers/button.h
+++ b/firmware/drivers/button.h
@@ -16,11 +16,14 @@
  * KIND, either express or implied.
  *
  ****************************************************************************/
+#ifndef _BUTTON_H_
+#define _BUTTON_H_
 
+#include <stdbool.h>
 #include "config.h"
 
 void button_init (void);
-int button_get (void);
+int button_get (bool block);
 
 /* Shared button codes */
 #define BUTTON_NONE             0x0000
@@ -52,3 +55,5 @@ int button_get (void);
 #define BUTTON_STOP             BUTTON_DOWN
 
 #endif
+
+#endif