bug fix for langevin dynamics