2 files changed, 2 insertions, 5 deletions

diff --git a/components/tollgate_core/src/tollgate_core.c b/components/tollgate_core/src/tollgate_core.c
index 3813258..666d2e0 100644
--- a/components/tollgate_core/src/tollgate_core.c
+++ b/components/tollgate_core/src/tollgate_core.c
@@ -45,11 +45,6 @@ esp_err_t tollgate_core_dns_start(esp_ip4_addr_t upstream_dns)
     return tollgate_core_dns_start_internal(s_ap_ip, upstream_dns);
 }
 
-void tollgate_core_dns_stop(void)
-{
-    tollgate_core_dns_stop_internal();
-}
-
 esp_err_t tollgate_core_process_payment(uint32_t client_ip, const char *token_str)
 {
     if (!s_platform || !token_str) return ESP_FAIL;
diff --git a/components/tollgate_core/src/tollgate_core_session.h b/components/tollgate_core/src/tollgate_core_session.h
index 95b5c48..444b9fa 100644
--- a/components/tollgate_core/src/tollgate_core_session.h
+++ b/components/tollgate_core/src/tollgate_core_session.h
@@ -2,6 +2,7 @@
 #define TOLLGATE_CORE_SESSION_H
 
 #include "esp_err.h"
+#include "tollgate_platform.h"
 #include <stdint.h>
 #include <stdbool.h>
 
@@ -19,6 +20,7 @@ typedef struct {
 } tg_session_t;
 
 esp_err_t tollgate_core_session_init(void);
+void tollgate_core_session_set_platform(const tollgate_platform_t *platform);
 
 tg_session_t *tollgate_core_session_create(uint32_t client_ip, uint64_t allotment_ms);
 tg_session_t *tollgate_core_session_create_bytes(uint32_t client_ip, uint64_t allotment_bytes);