diff --git a/src/wifi.h b/src/wifi.h
index c15e6fe..b0643e9 100644
--- a/src/wifi.h
+++ b/src/wifi.h
@@ -35,7 +35,7 @@ namespace Wifi {
             }
         );
 
-    void setup(Scheduler& ts, void(*onConnected)() = nullptr) {
+    void setup(Scheduler& ts, void(*onConnected)() = nullptr, void(*onDisconnected)() = nullptr) {
         stationConnectedHandler = WiFi.onStationModeGotIP(
             [onConnected](const WiFiEventStationModeGotIP& e) {
                 Serial.println("Connected to network.");
@@ -45,9 +45,10 @@ namespace Wifi {
             });
 
         stationDisconnectedHandler = WiFi.onStationModeDisconnected(
-            [](const WiFiEventStationModeDisconnected& e) {
+            [onDisconnected](const WiFiEventStationModeDisconnected& e) {
                 Serial.println("Disconnected from network.");
                 tReconnect.enableIfNot();
+                if (onDisconnected) onDisconnected();
             });