订阅到期后删除角色

Question

我有两种不同类型的订阅（基本订阅和加号订阅），并且根据所购买的订阅（第三方插件）分配了不同的角色。因此，当购买基本版时，他们仅被分配了基本角色，如果他们购买了加号，则将被分配加号。问题在于WooCommerce订阅只能在订阅到期时删除一个角色并分配一个角色。因此，当基本角色过期时，将删除基本角色，并从默认功能中分配过期角色。当加号过期时，将分配过期角色，但不会删除加号角色。我需要删除该加号角色。我该怎么办？

我正在尝试做类似的事情，但是我不确定自己在做什么吗？

add_action( 'woocommerce_subscription_status_canceled', 
'subscription_plus_removal' );

function subscription_plus_removal( $user_id, $subscription_key ) {
    $sub= wcs_get_subscription_from_key( $subscription_key );
    $user = new WP_User( $sub->user_id );

    // Remove role
    $user->remove_role( 'subscription_plus' );
}