diff --git a/TouchGFX/target/TouchGFXHAL.cpp b/TouchGFX/target/TouchGFXHAL.cpp
index c012ba4..2f6b546 100644
--- a/TouchGFX/target/TouchGFXHAL.cpp
+++ b/TouchGFX/target/TouchGFXHAL.cpp
@@ -24,6 +24,9 @@
 
 /* USER CODE BEGIN TouchGFXHAL.cpp */
 
+#include "STWButtonController.hpp"
+STWButtonController stwBC;
+
 using namespace touchgfx;
 
 void TouchGFXHAL::initialize()
@@ -35,6 +38,7 @@ void TouchGFXHAL::initialize()
     // Please note, HAL::initialize() must be called to initialize the framework.
 
     TouchGFXGeneratedHAL::initialize();
+    setButtonController(&stwBC);
 }
 
 /**