Code modification

Small code modification, no functional changes.