1 files changed, 5 insertions, 2 deletions
diff --git a/platformio/common/libs/mqtt_module_ota/homekit/mqtt/module/ota.h b/platformio/common/libs/mqtt_module_ota/homekit/mqtt/module/ota.h
index 53613c3..df4f7ce 100644
--- a/platformio/common/libs/mqtt_module_ota/homekit/mqtt/module/ota.h
+++ b/platformio/common/libs/mqtt_module_ota/homekit/mqtt/module/ota.h
@@ -57,11 +57,14 @@ private:
 public:
     MqttOtaModule() : MqttModule(0, true, true) {}
 
-    void init(Mqtt& mqtt) override;
+    void onConnect(Mqtt& mqtt) override;
+	void onDisconnect(Mqtt& mqtt, espMqttClientTypes::DisconnectReason reason) override;
+
     void tick(Mqtt& mqtt) override;
+	
     void handlePayload(Mqtt& mqtt, String& topic, uint16_t packetId, const uint8_t *payload, size_t length, size_t index, size_t total) override;
     void handleOnPublish(uint16_t packetId) override;
-    void handleOnDisconnect(espMqttClientTypes::DisconnectReason reason) override;
+
     inline bool isReadyToRestart() const {
         return ota.readyToRestart;
     }