diff --git a/cherab/compass/math/profile/electron.pxd b/cherab/compass/math/profile/electron.pxd
index b1cee70..629f10b 100644
--- a/cherab/compass/math/profile/electron.pxd
+++ b/cherab/compass/math/profile/electron.pxd
@@ -1,4 +1,4 @@
-from raysect.core.math.function cimport Function1D
+from raysect.core.math.function.float cimport Function1D
 
 
 cdef class ModifiedTanhGaussian(Function1D):
diff --git a/cherab/compass/math/profile/electron.pyx b/cherab/compass/math/profile/electron.pyx
index b07b4f4..8604144 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 raysect.core.math.function.float cimport Function1D
 
 from libc.math cimport exp