Please add a test for the legend too.