A slam bid, made, but sub-optimally

A stretch to reverse here, a pushy inquiry there saw me in slam. My partner correctly critiqued my play. But I don’t think you played it optimally. After you win the second trick, isn’t it best to play SA and ruff a spade high, then play a trump to dummy? If trumps are 4-0, you’ll still make if spades are 3-3, but if trumps are not 4-0, you can handle a 4-2 spade break by ruffing spades high twice more, getting back to dummy once more in trumps and the other time by ruffing the second round of diamonds. And if spades turn out to be 5-1, you’ll still survive if trumps are 2-2 and East is the one short in spades because you can ruff 3 red cards in dummy.