[DM][DocumentBaseline] Fix model