Forgot to add in extract QueryInvocation class