Fixed a bug in legend - missed a break