little code cleaning.