Wrong icon in angular stepper mat

I'm trying to put an edit icon when the state is address, but the icon shown is from the first state.

I've tried:

First step:

<ng-template matStepperIcon="edit">
    <mat-icon>home</mat-icon>
</ng-template>

<mat-step label="Antes de começar..." state="home">

..........

Second step

<ng-template matStepperIcon="address">
    <mat-icon>edit</mat-icon>
</ng-template>

<mat-step label="Dados do seu pet..." [stepControl]="secondFormGroup" state="address">    
        <div>
            <button mat-button matStepperPrevious>Back</button>
            <button mat-button matStepperNext>Next</button>
        </div>

Here's the result:

The second step should be the edit icon but it has the first step icon. Why?