Навесні 1944 р. після повернення з госпіталю льотчик був призначений заступником командира з ВСР 100-го ГВАП. Брав участь у боях під Яссами та Сандомиром. Знову набравши бойову форму, результативно боровся в боях Львівсько-Сандомирської операції, збив 6 німецьких літаків. У березні 1945 р. гвардії капітан Бабак став командиром уславленого «покришкінського» 16-го ГВАП. А через кілька днів у парі з молодим льотчиком вилетів на штурмівку в район міста Лаубаш. Мотор в літаку Бабака загорівся над ціллю, і він був змушений вистрибнути з парашутом.