diff --git a/controller/Controller.lua b/controller/Controller.lua
index db4790b..f7da7af 100644
--- a/controller/Controller.lua
+++ b/controller/Controller.lua
@@ -29,11 +29,11 @@ end
 
 function Controller:doAction(id, message, protocol)
     self:sendMessage(id, message, protocol)
-    local id2, message2
+    local id2, message2, protocol2
     repeat
-        id2, message2 = rednet.receive()
+        id2, message2, protocol2 = rednet.receive()
     until id2 == id
-    return message2 == "confirm", message2
+    return protocol2 == "confirm", message2
 end
 
 function Controller:setSlaveID(id)