Fixes bug in shortcode