В октябре 1944 года она вошла в особую авиагруппу 17-й воздушной армии генерал-майора авиации А. Н. Витрука, которая поддерживала войска Народно-освободительной армии Югославии С 17 октября она была перебазирована с болгарских аэродромов в Югославию на аэродром «Бела Црква» и вошла в подчинение командованию Народно-освободительной армии Югославии (НОАЮ). В том же месяце она принимала участие в Белградской наступательной операции. По освобождении Белграда в учебных центрах дивизии была организована подготовка авиационных кадров НОАЮ. За короткий период авиагруппе генерал-майора авиации А. Н. Витрука было подготовлено 140 югославских пилотов-истребителей и 145 штурмовиков на самолетах По-2 и Ил-2, 400 человек инженерно-технического состава, сформированы две югославские авиадивизии.