diff --git a/cherab/compass/math/profile/electron.pyx b/cherab/compass/math/profile/electron.pyx
index b07b4f4..26efffa 100644
--- a/cherab/compass/math/profile/electron.pyx
+++ b/cherab/compass/math/profile/electron.pyx
@@ -1,4 +1,4 @@
-from raysect.core.math.function cimport Function1D
+from cherab.core.math.function cimport Function1D
 
 from libc.math cimport exp