I'm OK with the check on the river for the reasons you gave. I would have reraised preflop.

He may only have Jx here and thought he could scare you with the second A coming. The river check would then be because he knows he's beat, another reason to check since he would fold to a raise, and in case you are WB you can avoid a trap.