Fix update query builder generation