diff --git a/Problem_ImageRegistration.py b/Problem_ImageRegistration.py
index fdf6396a8dc1e490f6abd6d46089f2d902bfd691..b2de6803d4265da180b70206b4826bed3ed2ce75 100644
--- a/Problem_ImageRegistration.py
+++ b/Problem_ImageRegistration.py
@@ -181,14 +181,22 @@ class ImageRegistrationProblem(Problem):
 
     def assemble_ener(self):
 
-        ener_form = 0.
+        ener = 0.
         for energy in self.energies:
-            ener_form += dolfin.Constant(energy.w) * energy.ener_form
-
-        ener = dolfin.assemble(
-            ener_form)
+            ener_ = dolfin.assemble(
+                energy.ener_form)
+            self.printer.print_var("ener_"+ener.name,ener_)
+            ener += energy.w * ener_
         #self.printer.print_var("ener",ener)
 
+        # ener_form = 0.
+        # for energy in self.energies:
+        #     ener_form += dolfin.Constant(energy.w) * energy.ener_form
+        #
+        # ener = dolfin.assemble(
+        #     ener_form)
+        # #self.printer.print_var("ener",ener)
+
         return ener