diff --git a/code/src/code.ino b/code/src/code.ino
index 27bb3e26..20341c35 100644
--- a/code/src/code.ino
+++ b/code/src/code.ino
@@ -547,11 +547,17 @@ void wifiLoop() {
             Serial.println();
         #endif
 
-        isMQTTMessage = true;
+        // Action to perform
+        if ((char)payload[0] == '0') {
+            isMQTTMessage = true;
+            switchRelayOff();
+        }
         if ((char)payload[0] == '1') {
+            isMQTTMessage = true;
             switchRelayOn();
-        } else {
-            switchRelayOff();
+        }
+        if ((char)payload[0] == '2') {
+            toggleRelay();
         }
         isMQTTMessage = false;