Error when trying to mock useMutation in Vitest

 it('should call onDeleteDistributionChannel when deleting a distribution channel', async () => {
    const user = userEvent.setup();
    const mockMutate = vi.fn((channelId, options) => {
      if (options?.onSuccess) {
        options.onSuccess();
      }
    });

    (useDeleteDistributionChannelMutation as Mock).mockReturnValue({
      mutate: mockMutate,
    });

    renderSurveyDistributionChannels({
      queries: { getDistributionChannels: mockDistributionChannel },
    });

    await waitForElementToBeRemoved(() => screen.getAllByRole('status'));

    const button = screen.getByRole('button', {
      name: `Slett kanalen ${singleDistributionChannel[0].type} på ${singleDistributionChannel[0].webhookUrl}`,
    });

    await user.click(button);

    await waitFor(() => expect(screen.getByRole('dialog')).toBeInTheDocument());

    expect(
      screen.getByRole('heading', {
        name: /fjern integrasjonen fra teamet/i,
      }),
    ).toBeInTheDocument();

    const deleteButton = screen.getByRole('button', { name: 'Slett' });

    await user.click(deleteButton);

    await waitFor(() => {
      expect(mockMutate).toHaveBeenCalledWith(
        singleDistributionChannel[0].id,
        expect.objectContaining({
          onSuccess: expect.any(Function),
        }),
      );
    });

    await waitFor(() => {
      expect(screen.queryByRole('dialog')).not.toBeInTheDocument();
    });
  });