Strengthen the updateCityCode function in query modules