WooCommerce Remove COD for User Role of Stockist

Remove COD for User Role Stockist

/* REMOVE COD FOR USER ROLE STOCKIST */
 
add_filter( 'woocommerce_available_payment_gateways', 'sow_checkout_restrictions', 10, 1 );
 
function sow_checkout_restrictions( $available_gateways ) {
    
    global $woocommerce;
 
    if (!empty(current_user_can('stockist') )) {
        unset( $available_gateways['cod'] );
    }
 
    return $available_gateways;
}