Use query cache aware trait