Can't define a specific build stage for a ImageFromDockerfile

Hello,
I'm using a container build from a Dockerfile with a multi-stage builds in my test:

public class SozuContainer <SELF extends SozuContainer<SELF>> extends GenericContainer<SELF> {
    public SozuContainer() {
        super(
            new ImageFromDockerfile()
                .withFileFromClasspath("Dockerfile", "sozu/Dockerfile")
        );
....
}     

And when I run my Junit test, the image build is the first stage of my multi-stage builds: builder.
Do you have a way to declare which target stage I want for my ImageFromDockerfile ?

Start with the ImageFromDockerfile entry point and the Dockerfile multi-stage build shown in the issue. Check how the JUnit test currently selects the built stage, then add coverage showing that a requested target stage is used and the default behavior remains intact.