diff --git a/lib/ai/formula_list.dart b/lib/ai/formula_list.dart
index 9689384..a9648e6 100644
--- a/lib/ai/formula_list.dart
+++ b/lib/ai/formula_list.dart
@@ -81,6 +81,11 @@ class _FormulaListState extends State<FormulaList> {
         builder: (context) => FormulaEditor(
           formula: formula,
           corpus: widget.corpus,
+          onSave: (updatedFormula){
+            setState((){
+              // THIS UPDATES THE FORMULA LIST
+            });
+          }
         ),
       ),
     );
diff --git a/lib/ai/formula_screen.dart b/lib/ai/formula_screen.dart
index 56758b6..9e92252 100644
--- a/lib/ai/formula_screen.dart
+++ b/lib/ai/formula_screen.dart
@@ -11,10 +11,10 @@ import 'unit_dropdown.dart';
 import 'formula_editor.dart';
 
 class FormulaScreen extends StatefulWidget {
-  final Formula formula;
+  Formula formula;
   final Corpus corpus;
 
-  const FormulaScreen({super.key, required this.formula, required this.corpus});
+  FormulaScreen({super.key, required this.formula, required this.corpus});
 
   @override
   State<FormulaScreen> createState() => _FormulaScreenState();
@@ -218,6 +218,7 @@ class _FormulaScreenState extends State<FormulaScreen> {
                       // Refresh the screen after saving
                       setState(() {
                         // The corpus has been updated, refresh the displayed formula
+                        widget.formula = updatedFormula;
                       });
                     },
                   ),