Small code update

N/A