fix matplotlib import behaviour of logp_plot