Fixing an imputation problem.