remove usage of prettyplotlib for compatibility