little correction
code cleaning