Fixed up penalised states/transistions.