xstate: Assign action in entry get a wrong state

Description Action in entry will get different state for assign action and non-assign action;

const machine = createMachine<ToggleContext, ToggleEvent>(
  {
    id: "machine",
    initial: "inactive",
    context: {
      count: 0
    },
    states: {
      inactive: {
        on: { TOGGLE: "active" },
        entry: ["assignEntry", "nonAssignEntry"]
      },
      active: {
        on: { TOGGLE: "inactive" },
        entry: ["assignEntry", "nonAssignEntry"]
      }
    }
  },
  {
    actions: {
      assignEntry: assign({
        count: (context, event, { state }) => {
          console.log("assign", state);
          return ++context.count;
        }
      }),
      nonAssignEntry: (context, event, { state }) => {
        console.log("non assign", state);
      }
    }
  }
);

Expected Result They should have same state value, I think the state should be the target state, not previous state;

Actual Result They have different state value:

Reproduction

• CodeSandbox template (TypeScript)

Additional context xstate version: 4.22.0