diff --git a/build/rocm/ci_build b/build/rocm/ci_build
index 9a2e9b19ec65..3519122dfbed 100755
--- a/build/rocm/ci_build
+++ b/build/rocm/ci_build
@@ -99,7 +99,7 @@ def dist_wheels(
 
     cmd.extend(mounts)
 
-    if os.isatty(sys.stdout.fileno()) and os.isatty(sys.stdin.fileno()):
+    if os.isatty(sys.stdout.fileno()):
         cmd.append("-it")
 
     # NOTE(mrodden): bazel times out without --init, probably blocking on a zombie PID
@@ -208,10 +208,12 @@ def test(image_name):
     cmd = [
         "docker",
         "run",
-        "-it",
         "--rm",
     ]
 
+    if os.isatty(sys.stdout.fileno()):
+        cmd.append("-it")
+
     # NOTE(mrodden): we need jax source dir for the unit test code only,
     # JAX and jaxlib are already installed from wheels
     # docker run fails when mounting ./