Fixed bug where gold mentions rather than 'correct' predicted mentions for attribute prediction