re-added a piece of code that got lost in an incorrect resolution of a merge conflict, works again
re-added a piece of code that got lost in an incorrect resolution of a merge conflict, works again

marian-nmt/marian-train,emjotde/amunmt,emjotde/amunmt,emjotde/amunmt,emjotde/amunn,emjotde/amunmt,marian-nmt/marian-train,marian-nmt/marian-train,emjotde/Marian,marian-nmt/marian-train,marian-nmt/marian-train,emjotde/amunn,emjotde/amunn,emjotde/Marian,emjotde/amunn