diff --git a/tests/test_split.py b/tests/test_split.py
index 93cf27b..d1e0a17 100644
--- a/tests/test_split.py
+++ b/tests/test_split.py
@@ -179,10 +179,8 @@ def Ffull(z, test):
         num_its_initial=10,
         num_its_per_step=4)
 
-    Tf = 1.0
-    while (float(t) < Tf):
-        if (float(t) + float(dt) > Tf):
-            dt.assign(Tf - float(t))
+    nsteps = 6
+    for _ in range(nsteps):
         imp_stepper.advance()
         imex_stepper.advance()
         t.assign(float(t) + float(dt))