diff --git a/unit_tests/utilities/test_zaza_utilities_openstack.py b/unit_tests/utilities/test_zaza_utilities_openstack.py
index e4abfb1..67cd417 100644
--- a/unit_tests/utilities/test_zaza_utilities_openstack.py
+++ b/unit_tests/utilities/test_zaza_utilities_openstack.py
@@ -1392,7 +1392,7 @@ class TestAsyncOpenstackUtils(ut_utils.AioTestCase):
 
         class AsyncContextManagerMock(test_mock):
             async def __aenter__(self):
-                yield model_mock
+                return self
 
             async def __aexit__(self, *args):
                 pass