In March 1943, he was given command of a rifle brigade, which was soon converted into the 76th Rifle Division, which he led during the Battle of Smolensk in the summer and early fall of 1943. Babayan was seriously wounded a third time during the battle, and after recovering took command of the 338th Rifle Division in spring 1944. He led the 338th in Operation Bagration and the Kaunas Offensive during the summer of 1944, but was relieved of command later that year. Babayan took command of the a brigade in the 1st Mechanized Corps in February 1945, which he led in the Vistula–Oder Offensive, the East Pomeranian Offensive, and the Battle of Berlin. Babayan was killed in action during the Battle of Berlin while personally leading an attack and posthumously awarded the title Hero of the Soviet Union.