Adding Query mock object because final is ugh