 He later returned to the United States and served as Assistant Division Commander, 3rd Armored Division, which was tasked with the training of replacement tank units during Korean War.Cole also served as Acting division commander from November to December 1952 and retired in mid-1953.Following his retirement from the Army after almost 36 years, Cole settled in Huntington Bay, New York, where he built a house.