diff --git a/tests/test_async_api.py b/tests/test_async_api.py
index de5eea8..ba9353f 100644
--- a/tests/test_async_api.py
+++ b/tests/test_async_api.py
@@ -84,7 +84,7 @@ async def _test():
                     description='Async integration test',
                     order=0,
                     policy=ChatsPolicy(
-                        event_type=DLPEventType.FILE_UPLOAD,
+                        event_types=[DLPEventType.FILE_UPLOAD],
                         action='prevent',
                         logging=LoggingStatus.ENABLED,
                         services_and_application=ServicesAndApplication(mode=SelectionMode.ALL),
@@ -317,7 +317,7 @@ async def _test():
                     description='Async toggle test',
                     order=0,
                     policy=ChatsPolicy(
-                        event_type=DLPEventType.PROMPT,
+                        event_types=[DLPEventType.PROMPT],
                         action='block',
                         logging=LoggingStatus.ENABLED,
                         services_and_application=ServicesAndApplication(mode=SelectionMode.ALL),
diff --git a/tests/test_unit_api.py b/tests/test_unit_api.py
index 8fa9269..6d60525 100644
--- a/tests/test_unit_api.py
+++ b/tests/test_unit_api.py
@@ -78,7 +78,7 @@ def test_genai_chats_rule_crud(self):
                 description='Integration test',
                 order=0,
                 policy=ChatsPolicy(
-                    event_type=DLPEventType.FILE_UPLOAD,
+                    event_types=[DLPEventType.FILE_UPLOAD],
                     action='prevent',
                     logging=LoggingStatus.ENABLED,
                     services_and_application=ServicesAndApplication(mode=SelectionMode.ALL),
@@ -292,7 +292,7 @@ def test_set_active_toggle(self):
                 description='Toggle test',
                 order=0,
                 policy=ChatsPolicy(
-                    event_type=DLPEventType.PROMPT,
+                    event_types=[DLPEventType.PROMPT],
                     action='block',
                     logging=LoggingStatus.ENABLED,
                     services_and_application=ServicesAndApplication(mode=SelectionMode.ALL),