fix a rare race condition in treeaction timeout detection

Forge.TreeWalker/src/TreeWalkerSession.cs

@@ -801,19 +801,6 @@ internal async Task PerformActionTypeBehavior(TreeNode treeNode, string treeNode
                 // Break out early if we would hit timeout before next retry.
                 if (actionTimeout != -1 && stopwatch.ElapsedMilliseconds + waitTime.TotalMilliseconds >= actionTimeout)
                 {
-                    // If the timeout is hit and the ContinuationOnTimeout flag is set, commit a new ActionResponse. 
-                    // with the status set to TimeoutOnAction and return.
-                    if (treeAction.ContinuationOnTimeout)
-                    {
-                        ActionResponse timeoutResponse = new ActionResponse
-                        {
-                            Status = "TimeoutOnAction"
-                        };
-
-                        await this.CommitActionResponse(treeActionKey, timeoutResponse).ConfigureAwait(false);
-                        return;
-                    }
-
                     break;
                 }
 
@@ -823,6 +810,22 @@ internal async Task PerformActionTypeBehavior(TreeNode treeNode, string treeNode
                 maxRetryCount--;
             }
 
+            if (actionTimeout != -1 && stopwatch.ElapsedMilliseconds + waitTime.TotalMilliseconds >= actionTimeout)
+            {
+                // If the timeout is hit and the ContinuationOnTimeout flag is set, commit a new ActionResponse. 
+                // with the status set to TimeoutOnAction and return.
+                if (treeAction.ContinuationOnTimeout)
+                {
+                    ActionResponse timeoutResponse = new ActionResponse
+                    {
+                        Status = "TimeoutOnAction"
+                    };
+
+                    await this.CommitActionResponse(treeActionKey, timeoutResponse).ConfigureAwait(false);
+                    return;
+                }
+            }
+
             // Retries are exhausted. Throw ActionTimeoutException with executeAction exception as innerException.
             throw new ActionTimeoutException(
                 string.Format(