Remove woocommerce gateways for $country, remove gateways for anything else

Unset woocommerce gateway for specific country

function lk_aussie_gateways( $available_gateways ) {

  global $woocommerce;

	if ($woocommerce->customer->get_country() == 'AU') {
		unset( $available_gateways['cod'] ); 
	}
	
	return $available_gateways; 

}

add_filter( 'woocommerce_available_payment_gateways', 'lk_aussie_gateways');


function lk_non_aussie_gateways( $available_gateways ) {

	global $woocommerce;

	if ($woocommerce->customer->get_country() <> 'AU' ) {
		unset ($available_gateways['paypal']);
		unset ($available_gateways['cheque']);

	}

	return $available_gateways; 
}

add_filter( 'woocommerce_available_payment_gateways', 'lk_non_aussie_gateways');