Tiny improvement to the fit method