Github Checks API waiting after build test

Hi,

I have a multi-stage workflow with build and test and then deployment.

It looks like the GH Checks API will wait for the entire workflow to finish, and not just the build and test. Before Checks API I was able to specify what stage I want to require for merging a PR.

What can I do here?

Screenshot: