diff --git a/kaldifeat/python/tests/test_fbank.py b/kaldifeat/python/tests/test_fbank.py
index 9e6f8a3..b95120c 100755
--- a/kaldifeat/python/tests/test_fbank.py
+++ b/kaldifeat/python/tests/test_fbank.py
@@ -6,10 +6,11 @@ import pickle
 from pathlib import Path
 
 import torch
-from utils import get_devices, read_ark_txt, read_wave
 
 import kaldifeat
 
+from .utils import get_devices, read_ark_txt, read_wave
+
 cur_dir = Path(__file__).resolve().parent
 
 
diff --git a/kaldifeat/python/tests/test_mfcc.py b/kaldifeat/python/tests/test_mfcc.py
index 33407b5..59f99a6 100755
--- a/kaldifeat/python/tests/test_mfcc.py
+++ b/kaldifeat/python/tests/test_mfcc.py
@@ -6,10 +6,11 @@ import pickle
 from pathlib import Path
 
 import torch
-from utils import get_devices, read_ark_txt, read_wave
 
 import kaldifeat
 
+from .utils import get_devices, read_ark_txt, read_wave
+
 cur_dir = Path(__file__).resolve().parent
 
 
diff --git a/kaldifeat/python/tests/test_plp.py b/kaldifeat/python/tests/test_plp.py
index 4f20452..ccc5e70 100755
--- a/kaldifeat/python/tests/test_plp.py
+++ b/kaldifeat/python/tests/test_plp.py
@@ -6,10 +6,11 @@ import pickle
 from pathlib import Path
 
 import torch
-from utils import get_devices, read_ark_txt, read_wave
 
 import kaldifeat
 
+from .utils import get_devices, read_ark_txt, read_wave
+
 cur_dir = Path(__file__).resolve().parent
 
 
diff --git a/kaldifeat/python/tests/test_spectrogram.py b/kaldifeat/python/tests/test_spectrogram.py
index 7e5a106..a080f7c 100755
--- a/kaldifeat/python/tests/test_spectrogram.py
+++ b/kaldifeat/python/tests/test_spectrogram.py
@@ -5,10 +5,10 @@
 import pickle
 from pathlib import Path
 
-from utils import get_devices, read_ark_txt, read_wave
-
 import kaldifeat
 
+from .utils import get_devices, read_ark_txt, read_wave
+
 cur_dir = Path(__file__).resolve().parent