Small code improvement.