If Stalin dies shortly before or after the end of WWII, it's entirely plausible that the Soviet government cooperates more extensively with the US in setting up the postwar world. There was immense gratitude to the US for Lend Lease and respect for its industrial might and wealth. A lot of Soviets hoped the postwar period would be better than the Stalinist one prior to the war. From 1945-1947, Stalin kept beating up his own diplomats for agreeing with the US so much and kept pushing them to not agree with American proposals.
Any Soviet official would want some kind of arrangement in Central Europe that afforded them protection and got some kind of loot, but without Stalin, it's entirely possible some arrangement could be worked out.
If Soviets allow the Polish government in exile to come back and allow free elections in Europe, agree to participate in Bretton woods and other postwar institutions, and the US makes available some economic aid and famine relief, then a beneficial relationship might develop.
Stalin needs to be dead though, and the sooner the better (even going back into 1941).