Revert "Update legend test"

This reverts commit f0cf33980a6310653fab41e3410400c3f44053a8.