Added a callback for the model to update in the query results code