Fix bug with collapse shortcode