Fix bug in shortcode