From b71e89596f5872202bc23062e6af6f72038b269b Mon Sep 17 00:00:00 2001
From: katdom13 <katdom13@gmail.com>
Date: Fri, 14 Jun 2024 18:11:45 +0800
Subject: [PATCH] Fix test failures on moto

---
 wagtail_storages/tests/test_wagtail_hooks.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/wagtail_storages/tests/test_wagtail_hooks.py b/wagtail_storages/tests/test_wagtail_hooks.py
index 6459ed7..1e83306 100644
--- a/wagtail_storages/tests/test_wagtail_hooks.py
+++ b/wagtail_storages/tests/test_wagtail_hooks.py
@@ -2,14 +2,14 @@
 
 from django.test import RequestFactory, TestCase, override_settings
 
-from moto import mock_s3
+from moto import mock_aws
 
 from wagtail_storages.factories import CollectionViewRestrictionFactory, DocumentFactory
 from wagtail_storages.tests.base import CreateBucket
 from wagtail_storages.wagtail_hooks import serve_document_from_s3
 
 
-@mock_s3
+@mock_aws
 class TestWagtailHooks(CreateBucket, TestCase):
     @override_settings(
         DEFAULT_FILE_STORAGE="django.core.files.storage.FileSystemStorage"