Yeah, I don't see playing 3-5-2 as a mistake.
Lazio chose to kill the game by simply matching up to Juve and neither side's individuals were able to assert themselves properly in the one-on-ones around the pitch to give an advantage one way or the other.
Given that the game was dead, there was no reason to think the odds weren't in Juve's favour given their superior individuals.