Slight code improvement.