I am trying to get a useMutation response and put it inside a state. Code:

const [createMessage] = useMutation(CREATE_MESSAGE, {
    onCompleted(data) {
     console.log(data.createMessage) //works fine
     setMessageId(data.createMessage.id)
    },
    variables: {
      message: {
        body,
        title,
      },
    },
  });

But for some reason setMessageId doesn't work. The state is still initial. What can I do to get a response?