diff --git a/src/one-body-sample.cc b/src/one-body-sample.cc
index 55add0c6412e54124ad367d62e80ce8b0460db79..759c0a1ed13d781a17b3360ae453c6a3f4cb7d78 100644
--- a/src/one-body-sample.cc
+++ b/src/one-body-sample.cc
@@ -235,6 +235,7 @@ int main(int argc, char *argv[]) {
     auto const &neumannFunction = functions.get("neumannCondition");
 
     for (size_t run = 1; run <= timesteps; ++run) {
+      double const time = h * run;
       if (parset.get<bool>("printProgress")) {
         std::cout << '*';
         std::cout.flush();
@@ -242,12 +243,12 @@ int main(int argc, char *argv[]) {
 
       {
         assemble_neumann<GridType, GridView, SmallVector, P1Basis>(
-            leafView, p1Basis, neumannNodes, b4, neumannFunction, h * run);
+            leafView, p1Basis, neumannNodes, b4, neumannFunction, time);
         stiffnessMatrix.mmv(u4, b4);
         // Apply Dirichlet condition
         for (int i = 0; i < finestSize; ++i)
           if (ignoreNodes[i].count() == dim) {
-            dirichletFunction.evaluate(h * run, u4_diff[i][0]);
+            dirichletFunction.evaluate(time, u4_diff[i][0]);
             u4_diff[i][0] /= refinement_factor;
           }