Minor code change