Any way to extend this function to assert that you don't have a legend? I think we had a few test cases like that.