Small code layout change