fix: use vanilla wait_for_task instead of a custom one

fix: use vanilla wait_for_task instead of a custom one #63