Move DB pool and helper functions into dedicated Database class

1 files changed, 1 insertions, 1 deletions

diff --git a/tests/unittest.py b/tests/unittest.py

index 295573bc46..fc856a574a 100644
--- a/tests/unittest.py
+++ b/tests/unittest.py
@@ -544,7 +544,7 @@ class HomeserverTestCase(TestCase):
         Add the given event as an extremity to the room.
         """
         self.get_success(
-            self.hs.get_datastore().simple_insert(
+            self.hs.get_datastore().db.simple_insert(
                 table="event_forward_extremities",
                 values={"room_id": room_id, "event_id": event_id},
                 desc="test_add_extremity",