diff --git a/tests/integ/sagemaker/conftest.py b/tests/integ/sagemaker/conftest.py
index fe7e7d61f8..421ef10b1d 100644
--- a/tests/integ/sagemaker/conftest.py
+++ b/tests/integ/sagemaker/conftest.py
@@ -14,16 +14,16 @@
 
 import base64
 import os
-import subprocess
-import shutil
-import pytest
-import docker
 import re
+import shutil
+import subprocess
 import sys
 
+import docker
+import pytest
 from docker.errors import BuildError
 
-from sagemaker.utils import sagemaker_timestamp, _tmpdir, sts_regional_endpoint
+from sagemaker.utils import _tmpdir, sagemaker_timestamp, sts_regional_endpoint
 
 REPO_ACCOUNT_ID = "033110030271"
 
@@ -68,7 +68,7 @@
     "RUN curl 'https://awscli.amazonaws.com/awscli-exe-linux-x86_64.zip' -o 'awscliv2.zip' \
         && unzip awscliv2.zip \
         && ./aws/install\n\n"
-    "RUN apt install sudo\n"
+    "RUN apt install -y sudo\n"
     "RUN useradd -ms /bin/bash integ-test-user\n"
     # Add the user to sudo group
     "RUN usermod -aG sudo integ-test-user\n"