Improved the logic to find a wrapper and assign the annotation_body class.