diff --git a/src/modules/auth/ui/views/sign-up-view.tsx b/src/modules/auth/ui/views/sign-up-view.tsx index d56075c..e976eeb 100644 --- a/src/modules/auth/ui/views/sign-up-view.tsx +++ b/src/modules/auth/ui/views/sign-up-view.tsx @@ -7,7 +7,7 @@ import { cn } from "@/lib/utils"; import { registerSchema, RegisterSchema } from "@/modules/auth/schemas"; import { useTRPC } from "@/trpc/client"; import { zodResolver } from "@hookform/resolvers/zod"; -import { useMutation } from "@tanstack/react-query"; +import { useMutation, useQueryClient } from "@tanstack/react-query"; import { Poppins } from "next/font/google"; import Link from "next/link"; import { useRouter } from "next/navigation"; @@ -22,11 +22,13 @@ const poppins = Poppins({ export const SignUpView = () => { const router = useRouter(); const trpc = useTRPC(); + const queryClient = useQueryClient(); const register = useMutation(trpc.auth.register.mutationOptions({ onError: (error) => { toast.error(error.message) }, - onSuccess: () => { + onSuccess: async () => { + await queryClient.invalidateQueries(trpc.auth.session.queryFilter()) router.push('/'); } }));