diff --git a/tests/laborious/activities/test_activities.py b/tests/laborious/activities/test_activities.py
index 8e6cd78..54fdd89 100644
--- a/tests/laborious/activities/test_activities.py
+++ b/tests/laborious/activities/test_activities.py
@@ -173,7 +173,7 @@ async def test_shutdown(
     mock_mlflow_init,
     mock_storage_init,
 ):
-    mock_opc_init.close = AsyncMock()
+    mock_opc_init.aclose = AsyncMock()
     postgres_config = {
         'host': 'localhost',
         'port': 5432,
@@ -221,7 +221,7 @@ async def test_shutdown(
     )
 
     await activities.shutdown()
-    mock_opc_init.close.assert_called_once()
+    mock_opc_init.aclose.assert_called_once()
     mock_storage_init.close.assert_called_once()
     mock_mlflow_init.close.assert_called_once()
     mock_gates_init.close.assert_called_once()