diff --git a/lib/ai/formula_screen.dart b/lib/ai/formula_screen.dart
index c8a4436..5b1c5bc 100644
--- a/lib/ai/formula_screen.dart
+++ b/lib/ai/formula_screen.dart
@@ -484,16 +484,9 @@ class _FormulaScreenState extends State<FormulaScreen> {
       }
 
       // Replace the current FormulaScreen with the new DerivedFormula screen
-      Navigator.pushReplacement(
-        context,
-        MaterialPageRoute(
-          builder: (context) => FormulaScreen(
-            formula: derivedFormula,
-            corpus: widget.corpus,
-            onSave: widget.onSave,
-          ),
-        ),
-      );
+      setState( (){
+        formula = derivedFormula;
+      });
     } catch (e, st) {
       errorHandler.notify(e,st);
     }