diff --git a/src/test/kotlin/org/opensearch/indexmanagement/indexstatemanagement/action/TransformActionIT.kt b/src/test/kotlin/org/opensearch/indexmanagement/indexstatemanagement/action/TransformActionIT.kt
index a054d94bc..08ba447f5 100644
--- a/src/test/kotlin/org/opensearch/indexmanagement/indexstatemanagement/action/TransformActionIT.kt
+++ b/src/test/kotlin/org/opensearch/indexmanagement/indexstatemanagement/action/TransformActionIT.kt
@@ -276,7 +276,7 @@ class TransformActionIT : IndexStateManagementRestTestCase() {
 
         // Change the start time so that the transform action will be attempted.
         updateManagedIndexConfigStartTime(managedIndexConfig)
-        waitFor {
+        waitFor(timeout = Instant.ofEpochSecond(60)) {
             assertEquals(
                 WaitForTransformCompletionStep.getJobCompletionMessage(transformId, indexName),
                 getExplainManagedIndexMetaData(indexName).info?.get("message")
@@ -312,7 +312,7 @@ class TransformActionIT : IndexStateManagementRestTestCase() {
 
         // Change the start time so that the transform action will be attempted.
         updateManagedIndexConfigStartTime(managedIndexConfig)
-        waitFor {
+        waitFor(timeout = Instant.ofEpochSecond(60)) {
             assertEquals(
                 WaitForTransformCompletionStep.getJobCompletionMessage(transformId, indexName),
                 getExplainManagedIndexMetaData(indexName).info?.get("message")
@@ -347,7 +347,7 @@ class TransformActionIT : IndexStateManagementRestTestCase() {
 
         // Change the start time so that the second transform action will be attempted.
         updateManagedIndexConfigStartTime(managedIndexConfig)
-        waitFor {
+        waitFor(timeout = Instant.ofEpochSecond(60)) {
             assertEquals(
                 WaitForTransformCompletionStep.getJobCompletionMessage(transformId, indexName),
                 getExplainManagedIndexMetaData(indexName).info?.get("message")