Merge from HEAD. Signed off by tmack.

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

diff --git a/PLC/Methods/UpdateAddressType.py b/PLC/Methods/UpdateAddressType.py
index c4e9d52..922c940 100644 (file)
--- a/PLC/Methods/UpdateAddressType.py
+++ b/PLC/Methods/UpdateAddressType.py
@@ -30,12 +30,13 @@ class UpdateAddressType(Method):
     def call(self, auth, address_type_id_or_name, address_type_fields):
         address_type_fields = dict(filter(can_update, address_type_fields.items()))
 
-        address_types = AddressTypes(self.api, [address_type_id_or_name]).values()
+        address_types = AddressTypes(self.api, [address_type_id_or_name])
         if not address_types:
             raise PLCInvalidArgument, "No such address type"
         address_type = address_types[0]
 
         address_type.update(address_type_fields)
         address_type.sync()
+       self.event_objects = {'AddressType': [address_type['address_type_id']]}
 
         return 1