small code revision