diff --git a/da/kzg_rs/fk20.py b/da/kzg_rs/fk20.py
index 97d1c90..f9eda8c 100644
--- a/da/kzg_rs/fk20.py
+++ b/da/kzg_rs/fk20.py
@@ -4,7 +4,7 @@ from eth2spec.deneb.mainnet import KZGProof as Proof, BLSFieldElement
 from eth2spec.utils import bls
 
 from da.kzg_rs.common import G1, BLS_MODULUS, PRIMITIVE_ROOT
-from da.kzg_rs.fft import fft, ifft, fft_g1, ifft_g1
+from da.kzg_rs.fft import fft, fft_g1, ifft_g1
 from da.kzg_rs.poly import Polynomial
 from da.kzg_rs.roots import compute_roots_of_unity
 from da.kzg_rs.utils import is_power_of_two
diff --git a/da/kzg_rs/test_fft.py b/da/kzg_rs/test_fft.py
index 93b4ef9..dc700c5 100644
--- a/da/kzg_rs/test_fft.py
+++ b/da/kzg_rs/test_fft.py
@@ -1,8 +1,8 @@
 from unittest import TestCase
 
-from da.kzg_rs.roots import compute_roots_of_unity
-from da.kzg_rs.common import BLS_MODULUS
-from fft import fft, ifft
+from .roots import compute_roots_of_unity
+from .common import BLS_MODULUS
+from .fft import fft, ifft
 
 
 class TestFFT(TestCase):