Lee helped plan the American airborne landings in Normandy, and trained to jump with his men. However, he was ordered to the United States several months before the invasion, having suffered either a heart attack or a stroke. He was replaced in command of the 101st by Major General Maxwell D. Taylor, who had previously commanded the 82nd Airborne Division Artillery, who went on to command the 101st until the end of the war.