diff --git a/tests/integ/test_auto_ml.py b/tests/integ/test_auto_ml.py
index 617ebae539..132b392160 100644
--- a/tests/integ/test_auto_ml.py
+++ b/tests/integ/test_auto_ml.py
@@ -170,6 +170,7 @@ def test_auto_ml_describe_auto_ml_job(sagemaker_session):
             },
             "TargetAttributeName": TARGET_ATTRIBUTE_NAME,
             "ContentType": "text/csv;header=present",
+            "ChannelType": "training",
         }
     ]
     expected_default_output_config = {
@@ -207,6 +208,7 @@ def test_auto_ml_attach(sagemaker_session):
             },
             "TargetAttributeName": TARGET_ATTRIBUTE_NAME,
             "ContentType": "text/csv;header=present",
+            "ChannelType": "training",
         }
     ]
     expected_default_output_config = {