Make diff-in-diff not return lm (non-serializable)