Small code cleaning.