fixed arguments to query_dataset callback