Small change in code