small bug in lr decay