Tests updated for new setAttribute method