From ac67b0484b283cc7a1c06316130ff6ec66e6f348 Mon Sep 17 00:00:00 2001
From: Chi Vinh Le <vinh@wikiwi.io>
Date: Fri, 2 Feb 2018 12:12:58 +0100
Subject: [PATCH] Better fix

---
 .../src/tabs/stream/components/FlagButton.js                | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/client/coral-embed-stream/src/tabs/stream/components/FlagButton.js b/client/coral-embed-stream/src/tabs/stream/components/FlagButton.js
index b1974db36..002297415 100644
--- a/client/coral-embed-stream/src/tabs/stream/components/FlagButton.js
+++ b/client/coral-embed-stream/src/tabs/stream/components/FlagButton.js
@@ -78,6 +78,8 @@ export default class FlagButton extends Component {
           return;
         }
         break;
+      case 2:
+        return this.closeMenu();
       default:
         throw new Error(`Unexpected step ${step}`);
     }
@@ -134,10 +136,6 @@ export default class FlagButton extends Component {
     }
 
     if (!failed) {
-      if (step === this.props.getPopupMenu.length - 1) {
-        this.closeMenu();
-        return;
-      }
       this.setState({ step: step + 1 });
     }
   };