Minor design tweek