Fix shortcode test failing