Add `wait_for_bucket_type`