Fix tests to handle the new space in the legend