diff --git a/library.json b/library.json
index 0162e48..cc7b8cd 100644
--- a/library.json
+++ b/library.json
@@ -1,6 +1,6 @@
 {
     "name": "Wifi",
-    "version": "1.0.1",
+    "version": "1.0.2",
     "description": "Helper classes for handling wifi connectiviy and OTA",
     "repository":
     {
diff --git a/src/wifi.h b/src/wifi.h
index 1b54e3c..b0643e9 100644
--- a/src/wifi.h
+++ b/src/wifi.h
@@ -33,9 +33,9 @@ namespace Wifi {
             currentPsk = WiFi.psk();
             tReconnect.enable();
             }
-            );
+        );
 
-    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.restartDelayed();
+                tReconnect.enableIfNot();
+                if (onDisconnected) onDisconnected();
             });