From 9b420253633ec826899f9c2a1d1488cea5d8447d Mon Sep 17 00:00:00 2001
From: Gareth Aneurin Tribello <garethtribello@Gareths-MacBook-Pro.local>
Date: Thu, 23 May 2024 12:09:04 +0100
Subject: [PATCH] Fixed a bug that makes units of time appear wrong in output
 file when MD code is not operating in ps

---
 src/core/PlumedMain.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/core/PlumedMain.cpp b/src/core/PlumedMain.cpp
index cdeba0472f..487650bd65 100644
--- a/src/core/PlumedMain.cpp
+++ b/src/core/PlumedMain.cpp
@@ -664,6 +664,7 @@ void PlumedMain::cmd(std::string_view word,const TypesafePtr & val) {
           ts = actionSet.selectWithLabel<ActionToPutData*>("timestep");
         }
         if( !ts->setValuePointer("timestep", val ) ) plumed_error();
+        ts->updateUnits( passtools.get() );
       }
       break;
       /* ADDED WITH API==2 */
@@ -674,6 +675,7 @@ void PlumedMain::cmd(std::string_view word,const TypesafePtr & val) {
         readInputLine("kBT: PUT CONSTANT PERIODIC=NO UNIT=energy", true);
         ActionToPutData* kb = actionSet.selectWithLabel<ActionToPutData*>("kBT");
         if( !kb->setValuePointer("kBT", val ) ) plumed_error();
+        kb->updateUnits( passtools.get() );
       }
       break;
       /* ADDED WITH API==3 */