Latent Covariates for Improved Causal Inference in Large-Scale Assessments