InvalidatateQuery problem

export const useAddUnitMutation = () => {
  const { queryClient } = useQueryClientInstance();

  return useMutation({
    mutationFn: (payload: TUnitPayload) => {
      return unitsAPI.createUnit(payload)
    },
    onSuccess: () => {
      toast.success('New unit added successfully');
      return queryClient.invalidateQueries(['units']);
    },
    onError: error => errorHandler(error),
  });
};

export const useUnits = (params?: ApiParams, options?: QueryOpt<Paginated<TUnit>>) => {
  return useQuery({
    queryKey: ['units', params],
    queryFn: () => unitsAPI.getUnits(params).then(res => res.data),
    ...options,
  });
};

 const handleUnitSubmit = async (payload: TUnitPayload) => {
    try {
      await create.mutateAsync(payload);

      reset();
      router.push(ROUTES.Units);
      return sleep();
    } catch (error) {
      errorHandler(error);
    }
  };