diff --git a/dune/tectonic/spatial-solving/tnnmg/functional.hh b/dune/tectonic/spatial-solving/tnnmg/functional.hh
index 4613c3aa533e8fa20ec730af569bb11475e2d5f0..9d08eca2b34cfb3115f69cd2375dd62b23514cc4 100755
--- a/dune/tectonic/spatial-solving/tnnmg/functional.hh
+++ b/dune/tectonic/spatial-solving/tnnmg/functional.hh
@@ -248,9 +248,11 @@ auto coordinateRestriction(const GlobalShiftedFunctional& f, const Index& i)
   Dune::MatrixVector::addProduct(ri, maxEig, origini);
 
   Aii = 0;
+  //print(Aii, "Aii: ");
   for (size_t j=0; j<Aii.N(); j++) {
-      Aii[j] = maxEig;
+      Aii[j][j] = maxEig;
   }
+  //print(Aii, "Aii: ");
 
   //print(*Aii_p, "Aii_p:");
   //print(ri, "ri:");
@@ -266,6 +268,8 @@ auto coordinateRestriction(const GlobalShiftedFunctional& f, const Index& i)
   dui -= origini;
 
   return Functional<decltype(Aii), LocalVector, decltype(phii), LocalLowerObstacle, LocalUpperObstacle, Range>(Aii, std::move(ri), phii, std::move(dli), std::move(dui));
+
+  //return ShiftedFunctional<decltype(Aii), LocalVector, decltype(phii), LocalLowerObstacle, LocalUpperObstacle, LocalVector, Range>(Aii, std::move(ri), phii, std::move(dli), std::move(dui), origini);
 }