Fix test which relied upon data-id attributes