There’s our next special tool invention!
Two-piece multi sized pump drive shaft extractor tool!

Has to be two sizes for 351 versus 302.
Actually, not sure that the end of the shafts are different, just that I’ve heard that the 351 engine shafts are slightly larger diameter.