Fix legend after previous commit