Correct test to reflect new markup