Shot: Application suffixes are not being handled correctly when testApplicationId is configured

Expected behaviour

If I run ./gradlew <Flavor><BuildType>ExecuteScreenshotTests -Precord I get all the screenshots recorded and downloaded from the device.

Actual behaviour

Build fails with this message:

Execution failed for task ‘:app:downloadScreenshots’. Nonzero exit value: 1

Steps to reproduce

This is my setup:

buildTypes {
        debug {
            applicationIdSuffix ".debug"
            versionNameSuffix "-debug"
        }
 }

productFlavors {
    dev {
        applicationIdSuffix ".dev"
        versionNameSuffix "-dev"
    }
}

Setup for Version 4.0.0:

shot {
    useComposer = true
}

Setup for Version 3.1.0:

shot {
    appId = "my.application.package.dev.debug"
    instrumentationTestTask = "testDevDebugComposer"
}

It also happens without using composer, by the way.

Attached the whole exception stacktrace gradleException.txt

Version of the library

3.1.0, 4.0.0