imputation bugfix