diff --git a/wfl/generate/md/__init__.py b/wfl/generate/md/__init__.py
index fafedb90..b36a6e89 100644
--- a/wfl/generate/md/__init__.py
+++ b/wfl/generate/md/__init__.py
@@ -202,7 +202,7 @@ def _get_temperature(atoms):
 
             if pressure_use is not None:
                 md_constructor = NPTBerendsen
-                stage_kwargs['pressure_au'] = pressure
+                stage_kwargs['pressure_au'] = pressure_use
                 stage_kwargs['compressibility_au'] = compressibility_au
                 stage_kwargs['taut'] = temperature_tau * fs
                 stage_kwargs['taup'] = pressure_tau * fs if pressure_tau is not None else temperature_tau * fs * 3