diff --git a/lib/ai/formula_screen.dart b/lib/ai/formula_screen.dart
index 8ef484a..c8a4436 100644
--- a/lib/ai/formula_screen.dart
+++ b/lib/ai/formula_screen.dart
@@ -472,13 +472,16 @@ class _FormulaScreenState extends State<FormulaScreen> {
     }
 
     try {
-      // Create a DerivedFormula with this input variable as output
-      final derivedFormula = DerivedFormula(
-        outputName: variable.name,
-        originalFormula: rootFormula
-      );
-
-
+      late final FormulaInterface derivedFormula;
+      if( variable.name == rootFormula.output.name) {
+        derivedFormula = rootFormula;
+      }
+      else {
+        derivedFormula = DerivedFormula(
+            outputName: variable.name,
+            originalFormula: rootFormula
+        );
+      }
 
       // Replace the current FormulaScreen with the new DerivedFormula screen
       Navigator.pushReplacement(