updated test for new legend fix