Of course, you are right.

I mean, the punishment was the occupation by forces which won the war. Americans and Russians had to control Germany for some time, as their current government could not continue for obvious reasons. The cruel part was giving control of half of the country to Soviets. BTW, worse was doing the same to Poland, which was victim, not the aggressor in this war, and other countries in similar situation.

But Germany were the baddies in WW2 and this was a part of their (cruel) punishment. How did Ukraine deserve this?

Or, maybe, USA and Russia want to get half of Ukraine each, for free?