TODO: check this is OK with all matplotlib versions