diff --git a/src/tests/test_asyncio.py b/src/tests/test_asyncio.py
index 016dc47f7..6818a081c 100644
--- a/src/tests/test_asyncio.py
+++ b/src/tests/test_asyncio.py
@@ -272,8 +272,8 @@ async def test_main(self, *, capsys: CaptureFixture) -> None:
         async def func() -> AsyncGenerator[int | None, float | None]:
             print("Initial")  # noqa: T201
             while True:
-                input_ = yield
-                output = round(ensure_not_none(input_))
+                input_ = ensure_not_none((yield))
+                output = round(input_)
                 if output >= 0:
                     print(f"Received {input_}, yielding {output}")  # noqa: T201
                     yield output
diff --git a/src/tests/test_functions.py b/src/tests/test_functions.py
index bdd808c3a..fa967a5d0 100644
--- a/src/tests/test_functions.py
+++ b/src/tests/test_functions.py
@@ -167,7 +167,7 @@ def test_main(self, *, capsys: CaptureFixture) -> None:
         def func() -> Generator[int | None, float | None, str]:
             print("Initial")  # noqa: T201
             while True:
-                input_ = yield
+                input_ = ensure_not_none((yield))
                 output = round(ensure_not_none(input_))
                 if output >= 0:
                     print(f"Received {input_}, yielding {output}")  # noqa: T201
diff --git a/src/utilities/__init__.py b/src/utilities/__init__.py
index 262a3af47..afd05d3a7 100644
--- a/src/utilities/__init__.py
+++ b/src/utilities/__init__.py
@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.54.0"
+__version__ = "0.54.1"