Corrections on heatbath