diff --git a/src/controller/python/chip/ChipDeviceCtrl.py b/src/controller/python/chip/ChipDeviceCtrl.py
index 4db95205179a46..4a38348575eccf 100644
--- a/src/controller/python/chip/ChipDeviceCtrl.py
+++ b/src/controller/python/chip/ChipDeviceCtrl.py
@@ -880,7 +880,7 @@ def SendGroupCommand(self, groupid: int, payload: ClusterObjects.ClusterCommand,
         return None
 
     async def WriteAttribute(self, nodeid: int,
-                             attributes: typing.List[typing.Tuple[int, ClusterObjects.ClusterAttributeDescriptor, int]],
+                             attributes: typing.List[typing.Tuple[int, ClusterObjects.ClusterAttributeDescriptor]],
                              timedRequestTimeoutMs: typing.Union[None, int] = None,
                              interactionTimeoutMs: typing.Union[None, int] = None, busyWaitMs: typing.Union[None, int] = None):
         '''
@@ -897,7 +897,7 @@ async def WriteAttribute(self, nodeid: int,
             to the XYZ attribute on the test cluster to endpoint 1
 
         Returns:
-            - [PyChipError] (list - one for each pth)
+            - [AttributeStatus] (list - one for each pth)
         '''
         self.CheckIsActive()