Good catch, I meant to add the attributes back too.