diff --git a/src/one-body-sample.cc b/src/one-body-sample.cc
index c90898532b7c431b7b06355d5afc9ed3cbb3fcc5..ed35729394f3ec0edec5a76140e7fe69051d8247 100644
--- a/src/one-body-sample.cc
+++ b/src/one-body-sample.cc
@@ -375,7 +375,7 @@ int main(int argc, char *argv[]) {
 
       { // Compute von Mises stress and write everything to a file
         auto *displacement =
-            new BasisGridFunction<P1Basis, VectorType>(p1Basis, u1);
+            new BasisGridFunction<P1Basis, VectorType>(p1Basis, u4);
         VonMisesStressAssembler<GridType> localStressAssembler(E, nu,
                                                                displacement);
         FunctionalAssembler<P0Basis>(p0Basis)
@@ -386,7 +386,7 @@ int main(int argc, char *argv[]) {
 
         auto displacement_ptr =
             Dune::make_shared<VTKBasisGridFunction<P1Basis, VectorType>>(
-                p1Basis, u1, "displacement");
+                p1Basis, u4, "displacement");
         auto vonmises_ptr =
             Dune::make_shared<VTKBasisGridFunction<P0Basis, CellVectorType>>(
                 p0Basis, vonMisesStress, "stress");