Your idea is better!
502nd Infantry? An uncle of mine was KIA in Korea while serving with the 507th PIR or 505 I think it was.