diff --git a/tests/test_version.py b/tests/test_version.py
index f658524..5aea599 100644
--- a/tests/test_version.py
+++ b/tests/test_version.py
@@ -1,4 +1,7 @@
-import toml
+try:
+    import toml
+except ImportError:
+    import tomlli as toml
 from pathlib import Path
 import dfschema