Recuperado da covid-19, Diego Souza volta a treinar