From 61a99e0e85e97cac08e661d5cf3c7b38ec528b67 Mon Sep 17 00:00:00 2001
From: Jim O'Donnell <james.odonnell@dtc.ox.ac.uk>
Date: Sun, 19 May 2024 12:14:00 +0100
Subject: [PATCH] Copy lab workflow changes to lab-fem

---
 app/pages/lab-fem/workflow.jsx | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/app/pages/lab-fem/workflow.jsx b/app/pages/lab-fem/workflow.jsx
index 797a4baaf8..d9ebc7a007 100644
--- a/app/pages/lab-fem/workflow.jsx
+++ b/app/pages/lab-fem/workflow.jsx
@@ -20,6 +20,7 @@ import TaskOptions from '../lab/workflow-components/task-options.jsx';
 import TaskEditor from '../lab/workflow-components/task-editor.jsx';
 import TaskPicker from '../lab/workflow-components/task-picker.jsx';
 import { isThisProjectUsingFEMLab, isWorkflowUsingJSONSubjects, FEM_LAB_PREVIEW_HOST } from './fem-lab-utilities.js';
+import removeTaskKeyFromWorkflow from '../lab/helpers/removeTaskKeyFromWorkflow.js';
 
 const DEMO_SUBJECT_SET_ID = process.env.NODE_ENV === 'production'
 ? '6' // Cats
@@ -848,8 +849,9 @@ class EditWorkflowPage extends Component {
         changes[`tasks.${taskKey}`] = undefined;
         this.props.workflow.update(changes);
       }
+      removeTaskKeyFromWorkflow(this.props.workflow, taskKey);
 
-      return this.updateFirstTask();
+      return this.updateFirstTask().save();
     }
   }
 };