Minor design fix