caret lm fix