TODO: check that this line is fine with any matplotlib version.