small design fix