I would play it the same. Too risky to try something tricky like checking river. Too likely he checks back AQ, AJ, ATs instead of betting them, and if I did check river, it would be with the intention of C/C because he certainly could bet value betting worse aces or missed spade draws. He does however, have a lot of hands in his range that beat you. He is probably calling your 3b and playing position with suited wheel aces and I’m sure he has at least 66 and 77 in his calling range also. 5’s not so sure. And 98s seems like a loooooong shot holding.

I think it is well played

Pio says check is higher EV than shove. It has to do with our flop size, which let villian have many combos of 2 pairs + that vs our shove villian doesnt have to defend AQ,AJ,AT.He prefers to hero call A8,A9,A2 instead.
However it is not huge difference, as well if we check we can check/fold more often than check/call.
Have to tweak tree in PIO a lot, because I doubt villian plays even close to it. Most case scenario over folding flop and turn which leaves us with one pair clear folds(86,97,T7,87), whether he is calling AQ,AJ often make our shoving EV go up which I wouldn't say is happening

I forgot to mention, villain had 67cc wich is a very loose defense from CO IMO.