Small code change.