diff --git a/dpti/relax.py b/dpti/relax.py
index ec01c8b..195d279 100755
--- a/dpti/relax.py
+++ b/dpti/relax.py
@@ -4,8 +4,8 @@
 import json
 import os
 
-from lib.lammps import get_last_dump
-from lib.utils import create_path, cvt_conf
+from dpti.lib.lammps import get_last_dump
+from dpti.lib.utils import create_path, cvt_conf
 
 
 def _gen_lammps_relax(conf_file, mass_map, model, pres, thermo_freq=100, dump_freq=100):
diff --git a/tests/test_relax.py b/tests/test_relax.py
new file mode 100644
index 0000000..8024114
--- /dev/null
+++ b/tests/test_relax.py
@@ -0,0 +1,13 @@
+import unittest
+
+from dpti import relax
+
+
+class TestRelaxImports(unittest.TestCase):
+    def test_package_module_imports(self):
+        """The relax module exposes its public task generator when installed."""
+        self.assertTrue(callable(relax.make_task))
+
+
+if __name__ == "__main__":
+    unittest.main()