Updating tests for selecting elements on the DOM