Update query-generator.js