En enero de 2017 fue convocado por el nuevo director técnico Tite para disputar un amistoso ante la selección de Colombia en homenaje a los futbolistas del Chapecoense que fallecieron en el vuelo 2933 de LaMia, lo que marcó el regreso de Diego a la "verdeamarelha".