After Gradle 9 migration the custom integration test fails with 'Cannot use shouldRunAfter'

[Volodymyr-Maiboroda-Imprivata]

We have the composite build and use the 0.17.20 version of the plugin.
The configuration we define looks like this:

dockerCompose {
    useComposeFiles = ["docker/docker-compose.yaml"]
    cit {
        useComposeFiles = ["docker/docker-compose.yaml"]
        startedServices = ["go-feature-flag"]
        isRequiredBy(componentIntegrationTest)
    }
}

It worked well on Gradle 7.6.4, but we have migrated to Gradle 9.2.1, and when I executed ./gradlew componentIntegrationTest, the build failed with:
Cannot use shouldRunAfter to reference tasks from another build.
When I remove the isRequiredBy() line, there is no issue, but the Docker Compose doesn't start.

Any idea how this can be fixed?

[Volodymyr-Maiboroda-Imprivata]

as a workaround, we applied this

dockerCompose {
    useComposeFiles = ["docker/docker-compose.yaml"]
    cit {
        useComposeFiles = ["docker/docker-compose.yaml"]
        startedServices = ["go-feature-flag"]
    }
}


componentIntegrationTest {
    dependsOn citComposeUp
    finalizedBy citComposeDown
}

but it's not ideal