Payment Gateway enable for few states in wocommerce

functions.php

function payment_gateway_disable_state( $available_gateways ) {
	global $woocommerce;

	$state = $woocommerce->customer->get_state();

	if ( isset( $available_gateways['cod'] ) &&  ! in_array( $state, array( 'AL', 'AK', 'CA' ) ) ) {
		unset( $available_gateways['cod'] );
	}

	return $available_gateways;
}

add_filter( 'woocommerce_available_payment_gateways', 'payment_gateway_disable_state' );