On diagonals of flag bundles