diff --git a/exafold/restraints/definitions.py b/exafold/restraints/definitions.py
index 5b7669d..ff60cf1 100644
--- a/exafold/restraints/definitions.py
+++ b/exafold/restraints/definitions.py
@@ -45,8 +45,8 @@
             # One for each parameter
             # in order of parameters list
             units=[
-                u.kilocalories_per_mole/u.angstrom**2,
-                u.angstrom
+                u.kilojoule_per_mole/u.angstrom**2,
+                u.nanometer,
             ],
         ),
     ),