Small functionality improvement.