Little design fix