diff --git a/tests/aci-e2e/e2e-aci_test.go b/tests/aci-e2e/e2e-aci_test.go
index 536ff779..3bcdea55 100644
--- a/tests/aci-e2e/e2e-aci_test.go
+++ b/tests/aci-e2e/e2e-aci_test.go
@@ -477,7 +477,7 @@ func TestContainerRunAttached(t *testing.T) {
 	})
 
 	t.Run("restart container", func(t *testing.T) {
-		res := c.RunDockerCmd("start", container)
+		res := c.RunDockerOrExitError("start", container)
 		//Flaky errors on restart : Code="ContainerGroupTransitioning" Message="The container group 'test-container' is still transitioning, please retry later."
 		if res.ExitCode != 0 && strings.Contains(res.Stderr(), `Code="ContainerGroupTransitioning"`) {
 			time.Sleep(3 * time.Second)