Update tests for the style element a little