Minor code change.