Fix data legend bug