diff --git a/tests/test_types.py b/tests/test_types.py
index 367a9c1..856dcf9 100644
--- a/tests/test_types.py
+++ b/tests/test_types.py
@@ -3,7 +3,7 @@
 import attr
 from astropy import units as u
 
-from py21cmsense import types as tp
+from py21cmsense import units as tp
 
 
 def test_vld_physical_type():