adjust test to check for Element class rather than just Object