COSMIT: typo in examples/imputation.py