Make sure the attributes.class is set properly