diff --git a/ui/src/flux/components/YieldFuncNode.tsx b/ui/src/flux/components/YieldFuncNode.tsx
index b642dfad54..1f35122bf9 100644
--- a/ui/src/flux/components/YieldFuncNode.tsx
+++ b/ui/src/flux/components/YieldFuncNode.tsx
@@ -37,11 +37,8 @@ class YieldFuncNode extends PureComponent<Props, State> {
     this.getData()
   }
 
-  public componentDidUpdate(prevProps) {
-    const {script: prevScript} = prevProps
-    const {script: currentScript} = this.props
-
-    if (prevScript !== currentScript) {
+  public componentDidUpdate(prevProps: Props) {
+    if (prevProps.script !== this.props.script) {
       this.getData()
     }
   }