- accept message_filter and return_fields

[plcapi.git] / PLC / Methods / UpdateMessage.py

diff --git a/PLC/Methods/UpdateMessage.py b/PLC/Methods/UpdateMessage.py
index 261eaeb..61ac458 100644 (file)
--- a/PLC/Methods/UpdateMessage.py
+++ b/PLC/Methods/UpdateMessage.py
@@ -31,12 +31,13 @@ class UpdateMessage(Method):
         message_fields = dict(filter(can_update, message_fields.items()))
 
         # Get message information
-        messages = Messages(self.api, [message_id]).values()
+        messages = Messages(self.api, [message_id])
         if not messages:
             raise PLCInvalidArgument, "No such message"
         message = messages[0]
 
         message.update(message_fields)
         message.sync()
+       self.object_ids = [message['message_id']]
 
         return 1