diff --git a/src/frontend/src/controllers/API/index.ts b/src/frontend/src/controllers/API/index.ts
index b2d21bffd..f0c36a215 100644
--- a/src/frontend/src/controllers/API/index.ts
+++ b/src/frontend/src/controllers/API/index.ts
@@ -111,7 +111,9 @@ export async function saveFlowToDatabase(newFlow: {
  * @returns {Promise<any>} The updated flow data.
  * @throws Will throw an error if the update fails.
  */
-export async function updateFlowInDatabase(updatedFlow: FlowType) {
+export async function updateFlowInDatabase(
+  updatedFlow: FlowType
+): Promise<FlowType> {
   try {
     const response = await axios.patch(`/api/v1/flows/${updatedFlow.id}`, {
       name: updatedFlow.name,