From 1aed0d51431734a44e8f969a4cd6d91af6de9988 Mon Sep 17 00:00:00 2001
From: Tamar Grey <tamar.grey@alteryx.com>
Date: Tue, 6 Feb 2024 09:25:26 -0500
Subject: [PATCH] Fix serialization tests by updating moto import

---
 featuretools/tests/entityset_tests/test_serialization.py      | 4 ++--
 .../tests/primitive_tests/test_feature_serialization.py       | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/featuretools/tests/entityset_tests/test_serialization.py b/featuretools/tests/entityset_tests/test_serialization.py
index c4c9081502..bc9ab36673 100644
--- a/featuretools/tests/entityset_tests/test_serialization.py
+++ b/featuretools/tests/entityset_tests/test_serialization.py
@@ -218,9 +218,9 @@ def test_to_pickle_id_none(tmp_path):
 @pytest.fixture
 def s3_client():
     _environ = os.environ.copy()
-    from moto import mock_s3
+    from moto import mock_aws
 
-    with mock_s3():
+    with mock_aws():
         s3 = boto3.resource("s3")
         yield s3
     os.environ.clear()
diff --git a/featuretools/tests/primitive_tests/test_feature_serialization.py b/featuretools/tests/primitive_tests/test_feature_serialization.py
index 8595ab504c..d2ca60af2c 100644
--- a/featuretools/tests/primitive_tests/test_feature_serialization.py
+++ b/featuretools/tests/primitive_tests/test_feature_serialization.py
@@ -210,9 +210,9 @@ def serialize_name_unchanged(original):
 @pytest.fixture
 def s3_client():
     _environ = os.environ.copy()
-    from moto import mock_s3
+    from moto import mock_aws
 
-    with mock_s3():
+    with mock_aws():
         s3 = boto3.resource("s3")
         yield s3
     os.environ.clear()