diff --git a/cherab/compass/math/profile/electron.pxd b/cherab/compass/math/profile/electron.pxd
index b1cee70..c3dad69 100644
--- a/cherab/compass/math/profile/electron.pxd
+++ b/cherab/compass/math/profile/electron.pxd
@@ -1,5 +1,4 @@
-from raysect.core.math.function cimport Function1D
-
+from cherab.core.math.function cimport Function1D
 
 cdef class ModifiedTanhGaussian(Function1D):
 
@@ -12,4 +11,4 @@ cdef class ModifiedTanhGaussian(Function1D):
 
     cdef double mtanh(self, double x)
 
-    cdef double core(self, double x)
\ No newline at end of file
+    cdef double core(self, double x)