Minor change to code