Fixing attribute issue