That's a post facto justification. Reading over the notes of the people doing the strategic planning for it all, it's quite clear they expected the war to continue. For example, there was a debate on if they should drop the nukes as they become available (which would have been a few a month), or if they should store them up and drop a whole lot on invasion day.
The Japanese had already fought on through the firebombing of Tokyo. That killed a comparable number of people to the atomic bombings. It just takes a lot more bombers to make it happen compared to dropping a nuke.