From 280ea0732bdc434dce8a3e7bb17d60301bb291cf Mon Sep 17 00:00:00 2001
From: Delmer Reed <delmer814+1@gmail.com>
Date: Wed, 20 Jun 2018 17:42:54 -0400
Subject: [PATCH] Remove yield nodes not in declarations

Co-authored-by: Iris Scholten <ischolten.is@gmail.com>
---
 ui/src/flux/containers/FluxPage.tsx | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/ui/src/flux/containers/FluxPage.tsx b/ui/src/flux/containers/FluxPage.tsx
index 17454e82a2..53e8305a13 100644
--- a/ui/src/flux/containers/FluxPage.tsx
+++ b/ui/src/flux/containers/FluxPage.tsx
@@ -549,7 +549,9 @@ export class FluxPage extends PureComponent<Props, State> {
           return `${declaration.name} = ${this.formatLastSource(s, isLast)}`
         }
 
-        const funcs = body.funcs.filter(f => f.id !== funcID)
+        const funcs = body.funcs.filter(
+          f => f.id !== funcID && f.id !== yieldNodeID
+        )
         const source = this.funcsToSource(funcs)
         return this.formatLastSource(source, isLast)
       })