Yes, you have it right. Japan collapsing politically would invite Soviet "aid" in the occupation of Japan. A partition like Germany was to be avoided. I just meant that you won't find anything from MacArthur clearly explaining it that way, as his retconned history, which became the accepted history, had a largely blameless imperial institution, so no justificatiom for its retention was really necessary.