Change test to only test element.