Very small code revision