Di Vaio set for Lazio return