From 1af39450275f4b2d2e82dc28b7718c0edf444ad0 Mon Sep 17 00:00:00 2001
From: Davor Runje <davor@airt.ai>
Date: Mon, 4 Mar 2024 16:34:20 +0100
Subject: [PATCH] bug fix

---
 tests/asyncapi/base/arguments.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/asyncapi/base/arguments.py b/tests/asyncapi/base/arguments.py
index 642ed24dbf..16386749d3 100644
--- a/tests/asyncapi/base/arguments.py
+++ b/tests/asyncapi/base/arguments.py
@@ -434,7 +434,7 @@ async def handle(user: descriminator): ...
 
         schema = get_app_schema(self.build_app(broker)).to_jsonable()
 
-        key = next(schema["components"]["messages"].keys())
+        key = next(iter(schema["components"]["messages"].keys()))
         assert key == IsStr(regex=r"test[\w:]*:Handle:Message")
         assert schema["components"] == {
             "messages": {