Fixed the attribute issue