small design flow fix