diff --git a/examples/state_dependent.py b/examples/state_dependent.py
index c30820d..e2cb4eb 100644
--- a/examples/state_dependent.py
+++ b/examples/state_dependent.py
@@ -6,7 +6,7 @@
 """
 
 
-import numpy
+import numpy as np
 
 from jitcdde import jitcdde, t, y
 
@@ -22,8 +22,9 @@
 
 DDE.integrate_blindly(0.01)
 
-data = []
-for time in [ DDE.t, *numpy.linspace(0.9,2,30) ]:
-	data.append([time, DDE.integrate(time)])
-numpy.savetxt("timeseries.dat", data)
+steps = np.linspace(0.9,2,30)
+data = np.empty((steps.size + 1, 2))
+for n, time in enumerate([ DDE.t, *steps ]):
+	data[n] = [time, DDE.integrate(time)[0]]
+np.savetxt("timeseries.dat", data)
 
diff --git a/tests/test_jitcdde.py b/tests/test_jitcdde.py
index 798c666..56693b8 100644
--- a/tests/test_jitcdde.py
+++ b/tests/test_jitcdde.py
@@ -9,8 +9,6 @@
 
 from jitcdde import (
 	UnsuccessfulIntegration,
-	_find_max_delay,
-	_get_delays,
     input,  # noqa: A004
 	jitcdde,
 	jitcdde_input,
@@ -19,6 +17,7 @@
 	test,
 	y,
 )
+from jitcdde._jitcdde import _find_max_delay, _get_delays
 
 
 dirname = os.path.dirname(__file__)