 During Nazi Germany's occupation, Vllamasi returned to Albania and became a member of the puppet government.He had an extremely difficult life during the communist rule of the Albanian Party of Labour, being imprisoned (1947–1956) and forced to work even in his old age.Vllamasi's memoirs, titled "Ballafaqime politike në Shqipëri (1899–1942) (Political Confrontations in Albania (1899–1942))", reveal him to be an open critic of other personalities like Luigj Gurakuqi and some of the Kosovar irredentists, and even the line of action followed by Noli and the opposition of the early 1920s .