A little code cleaning.