1 files changed, 4 insertions, 2 deletions
diff --git a/src/displayapp/screens/BleIcon.h b/src/displayapp/screens/BleIcon.h
index c1398d2a..d7410eae 100644
--- a/src/displayapp/screens/BleIcon.h
+++ b/src/displayapp/screens/BleIcon.h
@@ -1,12 +1,14 @@
 #pragma once
 
+#include "components/ble/BleController.h"
+
 namespace Pinetime {
   namespace Applications {
     namespace Screens {
       class BleIcon {
       public:
-        static const char* GetIcon(bool isConnected);
+        static const char* GetIcon(Pinetime::Controllers::Ble::ConnectStates state);
       };
     }
   }
-}
-\ No newline at end of file
+}