FIX: Fix the imputation example
Merge pull request #3603 from MechCoder/fix_imputation_example

[MRG] FIX: Fix the imputation example