diff --git a/tests/unit_tests/test_lattice_discretization.py b/tests/unit_tests/test_lattice_discretization.py
index 483cc111b8..b21d7e13e0 100644
--- a/tests/unit_tests/test_lattice_discretization.py
+++ b/tests/unit_tests/test_lattice_discretization.py
@@ -1,5 +1,5 @@
 
-from tests.unit_tests.test_lattice import rlat2
+from tests.unit_tests.test_lattice import zr, pincell1, pincell2, rlat2
 
 
 def test_discretization_degenerate(rlat2):