diff --git a/ui/src/flux/components/YieldFuncNode.tsx b/ui/src/flux/components/YieldFuncNode.tsx
index b420c04e35..1f35122bf9 100644
--- a/ui/src/flux/components/YieldFuncNode.tsx
+++ b/ui/src/flux/components/YieldFuncNode.tsx
@@ -37,6 +37,12 @@ class YieldFuncNode extends PureComponent<Props, State> {
     this.getData()
   }
 
+  public componentDidUpdate(prevProps: Props) {
+    if (prevProps.script !== this.props.script) {
+      this.getData()
+    }
+  }
+
   public render() {
     const {func} = this.props
     const {data} = this.state