Following the fall of Budapest, Vajna attempted to escape to Western Europe but was captured by units of the United States Army along with other members of the government. He was later tried in Budapest by a so-called people's tribunal and sentenced to death for war crimes, crimes against humanity and treason. He was hanged in 1946 in Budapest on the same day as Ferenc Szálasi, Károly Beregfy and József Gera.