Fix shortcode loading