small code adjustment