diff --git a/tests/integration/package/test_package_command_image.py b/tests/integration/package/test_package_command_image.py
index 5aac7e59c5..8c5b36649d 100644
--- a/tests/integration/package/test_package_command_image.py
+++ b/tests/integration/package/test_package_command_image.py
@@ -61,8 +61,9 @@ def test_package_template_without_image_repository(self, template_file):
             process.kill()
             raise
         process_stderr = stderr.strip()
-
-        self.assertIn("Error: Missing option '--image-repository'", process_stderr.decode("utf-8"))
+        self.assertIn(
+            "Error: Missing option '--image-repositories', '--image-repository'", process_stderr.decode("utf-8")
+        )
         self.assertEqual(2, process.returncode)
 
     @parameterized.expand(
@@ -189,7 +190,9 @@ def test_package_template_and_s3_bucket(self, template_file):
         process_stderr = stderr.strip()
 
         self.assertEqual(2, process.returncode)
-        self.assertIn("Error: Missing option '--image-repository'", process_stderr.decode("utf-8"))
+        self.assertIn(
+            "Error: Missing option '--image-repositories', '--image-repository'", process_stderr.decode("utf-8")
+        )
 
     @parameterized.expand(["aws-serverless-application-image.yaml"])
     def test_package_template_with_image_function_in_nested_application(self, template_file):