Al renunciar Oribe a la presidencia el 24 de octubre de 1838 y retirarse a Buenos Aires, Antonio Díaz lo acompañó, reasumiendo posteriormente su cargo en el gobierno "del Cerrito" cuando Oribe puso sitio a Montevideo, en 1843.