如何让这个wp_query在短代码中工作？

// 在主题的functions.php文件中注册短代码
function custom_query_shortcode($atts) {
    // 解析短代码参数
    $atts = shortcode_atts(array(
        'category' => 'uncategorized',
        'posts_per_page' => 5,
    ), $atts);

    // 设置wp_query参数
    $query_args = array(
        'category_name' => $atts['category'],
        'posts_per_page' => $atts['posts_per_page'],
    );

    // 执行wp_query查询
    $query = new WP_Query($query_args);

    // 输出查询结果
    if ($query->have_posts()) {
        $output = '<ul>';
        while ($query->have_posts()) {
            $query->the_post();
            $output .= '<li><a href="' . get_permalink() . '">' . get_the_title() . '</a></li>';
        }
        $output .= '</ul>';
    } else {
        $output = '没有找到相关文章。';
    }

    // 重置wp_query
    wp_reset_postdata();

    return $output;
}
add_shortcode('custom_query', 'custom_query_shortcode');