diff --git a/acapy_wallet_groups_plugin/v1_0/routes.py b/acapy_wallet_groups_plugin/v1_0/routes.py
index 094118d..3ec6db2 100644
--- a/acapy_wallet_groups_plugin/v1_0/routes.py
+++ b/acapy_wallet_groups_plugin/v1_0/routes.py
@@ -31,7 +31,6 @@
     wallet_update,
 )
 from aries_cloudagent.multitenant.base import BaseError, BaseMultitenantManager
-from aries_cloudagent.storage.base import DEFAULT_PAGE_SIZE
 from aries_cloudagent.storage.error import StorageError, StorageNotFoundError
 from aries_cloudagent.utils.endorsement_setup import (
     attempt_auto_author_with_endorser_setup,
diff --git a/poetry.lock b/poetry.lock
index 8eed4e4..fbbbcd1 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -285,7 +285,7 @@ didcommv2 = ["didcomm-messaging (>=0.1.1a0,<0.2.0)"]
 type = "git"
 url = "https://github.com/didx-xyz/aries-cloudagent-python.git"
 reference = "1.0.0b1"
-resolved_reference = "73b76839983b99e25aff35d63e23ede6f580d4ea"
+resolved_reference = "485ad5e8bfc98f77842400140d613d73a0bae26b"
 
 [[package]]
 name = "async-timeout"
diff --git a/tests/test_routes.py b/tests/test_routes.py
index 9199105..099911b 100644
--- a/tests/test_routes.py
+++ b/tests/test_routes.py
@@ -120,7 +120,7 @@ async def test_wallets_list(self):
                 ),
             ]
             mock_wallet_record.query = AsyncMock()
-            mock_wallet_record.query.return_value = [wallets[2], wallets[0], wallets[1]]
+            mock_wallet_record.query.return_value = wallets
 
             await test_module.wallets_list(self.request)
             mock_response.assert_called_once_with(