diff --git a/src/ancs_ble_client.h b/src/ancs_ble_client.h
index 37cd2e4..6faaf4d 100644
--- a/src/ancs_ble_client.h
+++ b/src/ancs_ble_client.h
@@ -3,7 +3,9 @@
 
 #include "ble_notification.h"
 
-#include "FreeRTOS.h" // For asynchronous tasks
+#include "freertos/FreeRTOS.h" // For asynchronous tasks
+#include "freertos/task.h"
+#include "Arduino.h"
 
 class BLEAddress;
 class BLERemoteCharacteristic;
@@ -65,4 +67,4 @@ private:
 	class BLERemoteCharacteristic* pControlPointCharacteristic;
 };
 
-#endif // ANCS_BLE_CLIENT_H_
\ No newline at end of file
+#endif // ANCS_BLE_CLIENT_H_
diff --git a/src/esp32notifications.cpp b/src/esp32notifications.cpp
index 4a6bd07..5cb58bf 100644
--- a/src/esp32notifications.cpp
+++ b/src/esp32notifications.cpp
@@ -108,11 +108,13 @@ bool BLENotifications::begin(const char * name) {
 	BLEDevice::setSecurityCallbacks(new NotificationSecurityCallbacks()); // @todo memory leak?
 	
 	startAdvertising();
+	return true;
 }
 
 bool BLENotifications::stop() {
 	ESP_LOGI(LOG_TAG, "stop()");
-	BLEDevice::deinit(false);	
+	BLEDevice::deinit(false);
+	return true;
 }