En la selección Diego utilizó la camiseta 18 durante el Sudamericano y la 17 en los Juegos Olímpicos, con la que convirtió su primer gol albiceleste.