Remove minor unwanted code change