diff --git a/docs/examples/richards_comparisonToCeiliaEtAl1990.py b/docs/examples/richards_comparisonToCeiliaEtAl1990.py
index 949031bd..ae7f5d1d 100644
--- a/docs/examples/richards_comparisonToCeiliaEtAl1990.py
+++ b/docs/examples/richards_comparisonToCeiliaEtAl1990.py
@@ -16,7 +16,8 @@ h = np.zeros(M.nC) + bc[0]
 
 
 def getFields(timeStep,method):
-    prob = Richards.RichardsProblem(M, mapping=E, timeStep=timeStep, timeEnd=360,
+    timeSteps = np.ones(360/timeStep)*timeStep
+    prob = Richards.RichardsProblem(M, mapping=E, timeSteps=timeSteps,
                                     boundaryConditions=bc, initialConditions=h,
                                     doNewton=False, method=method)
     return prob.fields(params['Ks'])