Whats the correct way to wait for `trpc.useQuery` to finish and then update state?

Like the title said, I have a case where I need to press a button 2x to update my state.

That happens because useEffect does not wait for useQuery to be finished. Is there a way to sync those two ?