diff --git a/packages/builder/src/components/integration/index.svelte b/packages/builder/src/components/integration/index.svelte
index f87883d991..a409a8efe1 100644
--- a/packages/builder/src/components/integration/index.svelte
+++ b/packages/builder/src/components/integration/index.svelte
@@ -67,7 +67,7 @@
       {/if}
     {:else if schema.type === QueryTypes.FLOW}
       <br />
-      {#if !query.fields.steps}
+      {#if query.fields.steps?.length == 0}
         <div class="controls">
           <Button
             secondary
@@ -91,7 +91,14 @@
               <div class="blockSection">
                 <div class="block-options">
                   Stage {index + 1}
-                  <ActionButton on:click={() => {}} icon="DeleteOutline" />
+                  <ActionButton
+                    on:click={() => {
+                      query.fields.steps = [
+                        ...query.fields.steps.splice(index, 1),
+                      ]
+                    }}
+                    icon="DeleteOutline"
+                  />
                 </div>
                 <Layout noPadding gap="S">
                   <div class="fields">
diff --git a/packages/server/src/integrations/mongodb.ts b/packages/server/src/integrations/mongodb.ts
index 5f969d76f3..77438756c4 100644
--- a/packages/server/src/integrations/mongodb.ts
+++ b/packages/server/src/integrations/mongodb.ts
@@ -52,7 +52,7 @@ module MongoDBModule {
       },
       aggregate: {
         type: QueryType.FLOW,
-      }
+      },
     },
     extra: {
       collection: {
@@ -327,11 +327,13 @@ module MongoDBModule {
         const db = this.client.db(this.config.db)
         const collection = db.collection(query.extra.collection)
         let response = {}
-        for await (const doc of collection.aggregate(query.steps.map(({ key, value }) => {
-          let temp:any = {}
-          temp[key] = JSON.parse(value.value)
-          return temp
-        }))) {
+        for await (const doc of collection.aggregate(
+          query.steps.map(({ key, value }) => {
+            let temp: any = {}
+            temp[key] = JSON.parse(value.value)
+            return temp
+          })
+        )) {
           response = doc
         }
         return response