restore PR#50 - see https://github.com/ctran/annotate_models/commit/cfeec96c9ca0fa5035b10be3d73e798cc4fc52f7
