Small design change