diff --git a/tests/test_chk.py b/tests/test_chk.py
index a5b7605..ce16441 100644
--- a/tests/test_chk.py
+++ b/tests/test_chk.py
@@ -4,7 +4,7 @@
 import numpy as np
 import pytest
 
-import w90io
+import wannier90io as w90io
 
 
 @pytest.mark.parametrize('example', ['example01', 'example02'])
diff --git a/tests/test_mmn.py b/tests/test_mmn.py
index 82cb61e..7db1395 100644
--- a/tests/test_mmn.py
+++ b/tests/test_mmn.py
@@ -2,7 +2,8 @@
 
 import numpy as np
 import pytest
-import w90io
+
+import wannier90io as w90io
 
 
 @pytest.mark.parametrize('example', ['example01', 'example02', 'example04'])