Minor layout design fix