Impute2 bug fix