Wedge fix based on behaviour not on mpl version