Update the query slightly