diff --git a/src/language/modules/SchemeModule.cpp b/src/language/modules/SchemeModule.cpp
index 84d263e6817151729a2ec3345aee46826b6adc03..98013190e42d1759aab0566462fe670dba2562d3 100644
--- a/src/language/modules/SchemeModule.cpp
+++ b/src/language/modules/SchemeModule.cpp
@@ -518,24 +518,29 @@ SchemeModule::SchemeModule()
 
                               ));
 
-  this->_addBuiltinFunction(
-    "energybalance",
-    std::make_shared<BuiltinFunctionEmbedder<std::tuple<
-      std::shared_ptr<const IDiscreteFunction>,
-      std::shared_ptr<const IDiscreteFunction>>(const std::shared_ptr<const IDiscreteFunction>&,
-                                                const std::shared_ptr<const IDiscreteFunction>&,
-                                                const std::shared_ptr<const IDiscreteFunction>&,
-                                                const std::shared_ptr<const IDiscreteFunction>&,
-                                                const std::vector<
-                                                  std::shared_ptr<const IBoundaryConditionDescriptor>>&)>>(
-      [](const std::shared_ptr<const IDiscreteFunction> lambdab, const std::shared_ptr<const IDiscreteFunction> mub,
-         const std::shared_ptr<const IDiscreteFunction> U, const std::shared_ptr<const IDiscreteFunction> source,
-         const std::vector<std::shared_ptr<const IBoundaryConditionDescriptor>>& bc_descriptor_list)
-        -> const std::tuple<std::shared_ptr<const IDiscreteFunction>, std::shared_ptr<const IDiscreteFunction>> {
-        return EnergyComputerHandler{lambdab, mub, U, source, bc_descriptor_list}.computeEnergyUpdate();
-      }
+  this->_addBuiltinFunction("energybalance",
+                            std::make_shared<BuiltinFunctionEmbedder<std::tuple<
+                              std::shared_ptr<const IDiscreteFunction>,
+                              std::shared_ptr<const IDiscreteFunction>>(const std::shared_ptr<const IDiscreteFunction>&,
+                                                                        const std::shared_ptr<const IDiscreteFunction>&,
+                                                                        const std::shared_ptr<const IDiscreteFunction>&,
+                                                                        const std::shared_ptr<const IDiscreteFunction>&,
+                                                                        const std::shared_ptr<const IDiscreteFunction>&,
+                                                                        const std::vector<std::shared_ptr<
+                                                                          const IBoundaryConditionDescriptor>>&)>>(
+                              [](const std::shared_ptr<const IDiscreteFunction> lambdab,
+                                 const std::shared_ptr<const IDiscreteFunction> mub,
+                                 const std::shared_ptr<const IDiscreteFunction> U,
+                                 const std::shared_ptr<const IDiscreteFunction> dual_U,
+                                 const std::shared_ptr<const IDiscreteFunction> source,
+                                 const std::vector<std::shared_ptr<const IBoundaryConditionDescriptor>>&
+                                   bc_descriptor_list) -> const std::tuple<std::shared_ptr<const IDiscreteFunction>,
+                                                                           std::shared_ptr<const IDiscreteFunction>> {
+                                return EnergyComputerHandler{lambdab, mub, U, dual_U, source, bc_descriptor_list}
+                                  .computeEnergyUpdate();
+                              }
 
-      ));
+                              ));
 
   this->_addBuiltinFunction("heat2",
                             std::make_shared<BuiltinFunctionEmbedder<
diff --git a/src/scheme/VectorDiamondScheme.hpp b/src/scheme/VectorDiamondScheme.hpp
index 9b56d86dbb3942c2ba51459c1f18394a56913ba8..261a9b56714bb087995b0bc4b6dd47503bfa2619 100644
--- a/src/scheme/VectorDiamondScheme.hpp
+++ b/src/scheme/VectorDiamondScheme.hpp
@@ -72,6 +72,7 @@ class EnergyComputerHandler
   EnergyComputerHandler(const std::shared_ptr<const IDiscreteFunction>& lambdab,
                         const std::shared_ptr<const IDiscreteFunction>& mub,
                         const std::shared_ptr<const IDiscreteFunction>& U,
+                        const std::shared_ptr<const IDiscreteFunction>& dual_U,
                         const std::shared_ptr<const IDiscreteFunction>& source,
                         const std::vector<std::shared_ptr<const IBoundaryConditionDescriptor>>& bc_descriptor_list);