Update drive to point proto message due to API changes