Make weak imputation optional