diff --git a/monkey/common/agent_configuration/agent_sub_configurations.py b/monkey/common/agent_configuration/agent_sub_configurations.py
index 9da21641c..4dbd9a3f2 100644
--- a/monkey/common/agent_configuration/agent_sub_configurations.py
+++ b/monkey/common/agent_configuration/agent_sub_configurations.py
@@ -8,16 +8,16 @@ class CustomPBAConfiguration:
     A configuration for custom post-breach actions
 
     Attributes:
-        linux_command (str): Command to run on Linux victim machines. If a file is uploaded,
-                             use this field to change its permissions, execute it, and/or delete
-                             it.
-                             Example: `chmod +x file.sh; ./file.sh; rm file.sh`
-        linux_filename (str): Name of the file to upload on Linux victim machines.
-        windows_command (str): Command to run on Windows victim machines. If a file is uploaded,
-                               use this field to change its permissions, execute it, and/or delete
-                               it.
-                               Example: `file.bat & del file.bat`
-        windows_filename (str): Name of the file to upload on Windows victim machines.
+        :param linux_command: Command to run on Linux victim machines. If a file is uploaded,
+                              use this field to change its permissions, execute it, and/or delete
+                              it.
+                              Example: `chmod +x file.sh; ./file.sh; rm file.sh`
+        :param linux_filename: Name of the file to upload on Linux victim machines.
+        :param windows_command: Command to run on Windows victim machines. If a file is uploaded,
+                                use this field to change its permissions, execute it, and/or delete
+                                it.
+                                Example: `file.bat & del file.bat`
+        :param windows_filename: Name of the file to upload on Windows victim machines.
     """
 
     linux_command: str