Hi,
If it's the 4 speed, I believe it is 40TE transmission. If you look at the attachments below, they provide the reassembly instructions for the valve body. I included all steps.
Additionally, it also shows the two check balls locations. I highlighted the heading on each of the relevant pages.
I hope this helps. Let me know how things turn out for you or if you have other questions,
Take care,
Joe
See pics below.
Images (Click to enlarge)
Dec 13, 2021 at 5:13 PM