Updated the test to keep up with the UI change in the previous commit.