Revert "Fix tests that were still using divs"

This reverts commit d50fdbfd216b0e56a5b382dd596c2ad41cfa56e6.
rever