diff --git a/platformio.ini b/platformio.ini
index 93fdda3..f5443ef 100644
--- a/platformio.ini
+++ b/platformio.ini
@@ -20,7 +20,7 @@ lib_deps =
 	jchristensen/Timezone@^1.2.4
 	adafruit/Adafruit Unified Sensor @ ^1.1.4
 	adafruit/Adafruit BME280 Library@^2.2.4
-	https://git.hodos.ro/libraries/wifi.git@^1.0.0
+	https://git.hodos.ro/libraries/wifi.git@^1.0.1
 	https://git.hodos.ro/libraries/ha-mqtt.git@^1.0.0
 build_flags = -D WIFI_ALWAYS_ON=1
 
diff --git a/src/esp_clock.cpp b/src/esp_clock.cpp
index f83af4e..0e65126 100644
--- a/src/esp_clock.cpp
+++ b/src/esp_clock.cpp
@@ -57,7 +57,13 @@ void setup() {
   digitalWrite(LED_BUILTIN, HIGH);
 
   Display::setup();
-  Ota::setup();
+  Ota::setup(
+      [] {
+          Display::displayText("load");
+          Mqtt::publishCleanupConfig();
+          delay(2000);
+          Mqtt::disconnect();
+      });
   Ntp::setup();
   Bme::setup();
   Mqtt::setup(&ts,