small code improvement