Expand AZ awareness to explicitly set the rack_id as an awareness attribute