Small design fix