diff --git a/dev-requirements.txt b/dev-requirements.txt
index bee78d4cc..8dbcac2ec 100644
--- a/dev-requirements.txt
+++ b/dev-requirements.txt
@@ -15,7 +15,7 @@ hydra-core
 ipython
 kfp==1.8.22
 mlflow-skinny
-moto==4.2.14
+moto~=5.0.8
 pyre-extensions
 pyre-check
 pytest
diff --git a/scripts/component_integration_tests.py b/scripts/component_integration_tests.py
index c4a512c3d..2eb3013d4 100755
--- a/scripts/component_integration_tests.py
+++ b/scripts/component_integration_tests.py
@@ -174,13 +174,9 @@ def _mock_aws_batch() -> None:
     ensure_network()
     os.environ.setdefault("MOTO_DOCKER_NETWORK_NAME", NETWORK)
 
-    from moto import mock_batch, mock_ec2, mock_ecs, mock_iam, mock_logs
+    from moto import mock_aws
 
-    mock_batch().__enter__()
-    mock_iam().__enter__()
-    mock_ec2().__enter__()
-    mock_ecs().__enter__()
-    mock_logs().__enter__()
+    mock_aws().__enter__()
 
     import boto3.session