Use query init callback