fix: revert d53c237

This seems to not be working as expected because `main` is also getting
cancelled.

[Cherry-picked c6c35f9][modified]

Author: Kordyjan