diff --git a/src/one-body-sample.cc b/src/one-body-sample.cc
index 9611eb140d077553c12e81bc86d4b0d437a7ce23..be42cd1d220a44095f9ab8499a27e6ae7a09c10a 100644
--- a/src/one-body-sample.cc
+++ b/src/one-body-sample.cc
@@ -238,7 +238,6 @@ int main(int argc, char *argv[]) {
       stiffnessMatrix.mmv(u3, b3);
 
       // {{{ Assemble terms for the nonlinearity
-
       std::vector<double> mu;
       mu.resize(grid.size(grid.maxLevel(), dim));
       std::fill(mu.begin(), mu.end(),
@@ -292,14 +291,14 @@ int main(int argc, char *argv[]) {
 
       u1 += u1_diff_new;
 
-      auto *displacement =
-          new BasisGridFunction<P1Basis, VectorType>(p1Basis, u1);
-      VonMisesStressAssembler<GridType> localStressAssembler(E, nu,
-                                                             displacement);
-      FunctionalAssembler<P0Basis>(p0Basis)
-          .assemble(localStressAssembler, vonMisesStress, true);
+      { // Compute von Mises stress and write everything to a file
+        auto *displacement =
+            new BasisGridFunction<P1Basis, VectorType>(p1Basis, u1);
+        VonMisesStressAssembler<GridType> localStressAssembler(E, nu,
+                                                               displacement);
+        FunctionalAssembler<P0Basis>(p0Basis)
+            .assemble(localStressAssembler, vonMisesStress, true);
 
-      {
         Dune::VTKWriter<GridType::LeafGridView> writer(leafView);
         std::string filename((boost::format("obs%d") % run).str());
         Dune::shared_ptr<VTKBasisGridFunction<P1Basis, VectorType>>