diff --git a/mqttbutton.lua b/mqttbutton.lua
index 757c318..99bead4 100644
--- a/mqttbutton.lua
+++ b/mqttbutton.lua
@@ -43,8 +43,8 @@ end
 
 function on_mqtt_disconnected(client)
     print("MQTT Disconnected")
-    client:close()
-    on_wifi_timer()
+    mqtt_client:close()
+    tmr.alarm(WIFI_TIMER_ID, 1000, 0, on_wifi_timer)
 end
 
 function on_wifi_connected()