summary refs log tree commit diff
path: root/tests/rest/client/test_shadow_banned.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/rest/client/test_shadow_banned.py')
-rw-r--r--tests/rest/client/test_shadow_banned.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/rest/client/test_shadow_banned.py b/tests/rest/client/test_shadow_banned.py
index b0c44af033..7d0e66b534 100644
--- a/tests/rest/client/test_shadow_banned.py
+++ b/tests/rest/client/test_shadow_banned.py
@@ -34,7 +34,7 @@ class _ShadowBannedBase(unittest.HomeserverTestCase):
         self.banned_user_id = self.register_user("banned", "test")
         self.banned_access_token = self.login("banned", "test")
 
-        self.store = self.hs.get_datastore()
+        self.store = self.hs.get_datastores().main
 
         self.get_success(
             self.store.set_shadow_banned(UserID.from_string(self.banned_user_id), True)