Shortly thereafter he was arrested and sentenced to 5-year imprisonment. However, in 1940 he was released among a few other general officers such as Gorbatov and Rokossovsky and reinstated in his previous position. Following adoption of General ranks in the Red Army, he was created Major General.
In August 1941, he took command of the 40th Army, which was trying to stop the advance of Guderian's 2nd Panzer Group to Moscow. After that, he was promoted to Lieutenant General.