Can we remove the `class` attribute here?