diff --git a/src/one-body-sample.cc b/src/one-body-sample.cc
index 427ec512cc7b35304d740b0599139298b737ab6f..769ecb8e96e4bbb39df16394a867f59ab7168fdd 100644
--- a/src/one-body-sample.cc
+++ b/src/one-body-sample.cc
@@ -249,8 +249,10 @@ int main(int argc, char *argv[]) {
         stiffnessMatrix.mmv(u, rhs);
         // Apply Dirichlet condition
         for (size_t i = 0; i < finestSize; ++i)
-          if (ignoreNodes[i].count() == dim)
+          if (ignoreNodes[i].count() == dim) {
             dirichletFunction.evaluate(time, u_diff[i][0]);
+            u_diff[i][0] /= timesteps;
+          }
 
         auto const state_fpi_max =
             parset.get<size_t>("solver.tnnmg.fixed_point_iterations");
diff --git a/src/one-body-sample.py b/src/one-body-sample.py
index cd0625563cbd731b6e1b7a60ba68a2b6222d6641..db96717227e1265df190e7a60bc83e13a2f473b2 100644
--- a/src/one-body-sample.py
+++ b/src/one-body-sample.py
@@ -11,14 +11,13 @@ class neumannCondition:
         else:
             return trd * (x - 2.0/3) + (fst + snd) * 1.0/3
 
-## FIXME: If we change time discretisation, we currently change the problem
 class dirichletCondition:
     def __call__(self, x):
-        return 2e-3
+        return 2
         # return 0
-        fst = 3e-4
-        snd = 5e-4
-        trd = 3e-4
+        fst = 3e-1
+        snd = 5e-1
+        trd = 3e-1
         if x < 1.0/5:
             return fst
         elif x < 3.0/5: