fix matplotlib example