Fixed the attribute issue

Fixed the attribute issue