If russia falls then kaliningrad should be given to Lithuania.

Because giving it to germany would just create an outrage across the EU for obvious reasons and Poland is significantly bigger country so giving it to Lithuania is more economical for everyone involved.