Fix test now that main section is a main element