machine.withConfig() returns a StateNode type instead of a StateMachine type, breaking event inference downstream

[VanTanev]

Description
StateMachine.withConfig() returns a StateNode instead of a StateMachine. This breaks interpreter inference.

import { createMachine, interpret } from 'xstate'

interface Context {}
type Event = { type: 'EVENT' }
const machine = createMachine<Context, Event>({})

const interpreter1: InterpreterFrom<typeof machine> = interpret(machine)

// @ts-expect-error Event object is not inferred properly for the Interpreter
const interpreter2: InterpreterFrom<typeof machine> = interpret(machine.withConfig({}))

Reproduction
https://codesandbox.io/s/machinewithcontext-returns-a-statenode-type-instead-of-a-statemachine-type-k2mxk

Fix?

export interface StateMachine<
  TContext,
  TStateSchema extends StateSchema,
  TEvent extends EventObject,
  TTypestate extends Typestate<TContext> = { value: any; context: TContext }
> extends StateNode<TContext, TStateSchema, TEvent, TTypestate> {
  id: string;
  states: StateNode<TContext, TStateSchema, TEvent>['states'];

  withConfig(
    options: Partial<MachineOptions<TContext, TEvent>>,
    context?: TContext
  ): StateMachine<TContext, TStateSchema, TEvent>;

  withContext(
    context: TContext
  ): StateMachine<TContext, TStateSchema, TEvent>
}

[VanTanev]

PR: #2240