minor design fix