fix reconstruction of word classes instead of target words