diff --git a/awsbatch-cli/tox.ini b/awsbatch-cli/tox.ini
index 36e087b2d5..cac9d46200 100644
--- a/awsbatch-cli/tox.ini
+++ b/awsbatch-cli/tox.ini
@@ -14,7 +14,6 @@ usedevelop =
     nocov: false
 deps =
     -rtests/requirements.txt
-    pytest-travis-fold
 commands =
     nocov: pytest -n auto -l -v --basetemp={envtmpdir} --html=report.html --ignore=src tests/
     cov: python setup.py clean --all build_ext --force --inplace
diff --git a/cli/tox.ini b/cli/tox.ini
index c5f9ead79e..827948717c 100644
--- a/cli/tox.ini
+++ b/cli/tox.ini
@@ -17,7 +17,6 @@ allowlist_externals =
     bash
 deps =
     -rtests/requirements.txt
-    pytest-travis-fold
 extras =
     awslambda
 commands =
diff --git a/tests/integration-tests/tests/iam/test_iam_image.py b/tests/integration-tests/tests/iam/test_iam_image.py
index 63acdddb9a..7f05a3122f 100644
--- a/tests/integration-tests/tests/iam/test_iam_image.py
+++ b/tests/integration-tests/tests/iam/test_iam_image.py
@@ -93,7 +93,7 @@ def _get_resources_with_image_resource(cfn_client, stack_name):
     image_resource_exists = False
     logging.info("Checking image resource")
     for resource in resources:
-        if resource["ResourceType"] == "AWS::ImageBuilder::Image":
+        if resource["ResourceType"] == "AWS::ImageBuilder::Image" and resource.get("PhysicalResourceId"):
             image_resource_exists = True
             logging.info("The image resource exists!")
             break