Fix AZ set state