diff --git a/src/main/java/org/jenkinsci/plugins/workflow/support/visualization/table/FlowGraphTable.java b/src/main/java/org/jenkinsci/plugins/workflow/support/visualization/table/FlowGraphTable.java
index 9496a4f1..a56675ab 100644
--- a/src/main/java/org/jenkinsci/plugins/workflow/support/visualization/table/FlowGraphTable.java
+++ b/src/main/java/org/jenkinsci/plugins/workflow/support/visualization/table/FlowGraphTable.java
@@ -262,7 +262,7 @@ public static class Row {
 
         private Row(FlowNode node) {
             this.node = node;
-            TimingAction act = node.getAction(TimingAction.class);
+            TimingAction act = node.getPersistentAction(TimingAction.class);
             if (act != null) {
                 this.startTimeMillis = act.getStartTime();
                 this.hasStartTime = true;
@@ -289,7 +289,7 @@ public String getDisplayName() {
                 // TODO make StepAtomNode.effectiveFunctionName into an API
                 return node.getDisplayFunctionName();
             } else if (node instanceof StepNode && node instanceof BlockStartNode) {
-                if (node.getAction(BodyInvocationAction.class) != null) {
+                if (node.getPersistentAction(BodyInvocationAction.class) != null) {
                     // TODO cannot access StepAtomNode.effectiveFunctionName from here
                     LinearBlockHoppingScanner scanner = new LinearBlockHoppingScanner();
                     scanner.setup(node);