Minor bug fix in shortcode