fix LM with last TRR step